|
1 |
| -import pytest |
2 | 1 | import re
|
| 2 | + |
3 | 3 | import numpy as np
|
| 4 | +import pytest |
4 | 5 | import scipy.sparse as sp
|
5 | 6 | from joblib import cpu_count
|
6 |
| -import re |
7 | 7 |
|
8 |
| -from sklearn.utils._testing import assert_almost_equal |
9 |
| -from sklearn.utils._testing import assert_array_equal |
10 |
| -from sklearn.utils._testing import assert_array_almost_equal |
11 | 8 | from sklearn import datasets
|
12 |
| -from sklearn.base import clone |
13 |
| -from sklearn.datasets import make_classification |
14 |
| -from sklearn.datasets import load_linnerud |
15 |
| -from sklearn.datasets import make_multilabel_classification |
16 |
| -from sklearn.datasets import make_regression |
17 |
| -from sklearn.ensemble import GradientBoostingRegressor, RandomForestClassifier |
| 9 | +from sklearn.base import ClassifierMixin, clone |
| 10 | +from sklearn.datasets import ( |
| 11 | + load_linnerud, |
| 12 | + make_classification, |
| 13 | + make_multilabel_classification, |
| 14 | + make_regression, |
| 15 | +) |
| 16 | +from sklearn.ensemble import ( |
| 17 | + GradientBoostingRegressor, |
| 18 | + RandomForestClassifier, |
| 19 | + StackingRegressor, |
| 20 | +) |
18 | 21 | from sklearn.exceptions import NotFittedError
|
19 |
| -from sklearn.linear_model import Lasso |
20 |
| -from sklearn.linear_model import LogisticRegression |
21 |
| -from sklearn.linear_model import OrthogonalMatchingPursuit |
22 |
| -from sklearn.linear_model import Ridge |
23 |
| -from sklearn.linear_model import SGDClassifier |
24 |
| -from sklearn.linear_model import SGDRegressor |
25 |
| -from sklearn.linear_model import LinearRegression |
| 22 | +from sklearn.impute import SimpleImputer |
| 23 | +from sklearn.linear_model import ( |
| 24 | + Lasso, |
| 25 | + LinearRegression, |
| 26 | + LogisticRegression, |
| 27 | + OrthogonalMatchingPursuit, |
| 28 | + Ridge, |
| 29 | + SGDClassifier, |
| 30 | + SGDRegressor, |
| 31 | +) |
26 | 32 | from sklearn.metrics import jaccard_score, mean_squared_error
|
| 33 | +from sklearn.model_selection import GridSearchCV, train_test_split |
27 | 34 | from sklearn.multiclass import OneVsRestClassifier
|
28 |
| -from sklearn.multioutput import ClassifierChain, RegressorChain |
29 |
| -from sklearn.multioutput import MultiOutputClassifier |
30 |
| -from sklearn.multioutput import MultiOutputRegressor |
| 35 | +from sklearn.multioutput import ( |
| 36 | + ClassifierChain, |
| 37 | + MultiOutputClassifier, |
| 38 | + MultiOutputRegressor, |
| 39 | + RegressorChain, |
| 40 | +) |
| 41 | +from sklearn.pipeline import make_pipeline |
31 | 42 | from sklearn.svm import LinearSVC
|
32 | 43 | from sklearn.tree import DecisionTreeClassifier
|
33 |
| -from sklearn.base import ClassifierMixin |
34 | 44 | from sklearn.utils import shuffle
|
35 |
| -from sklearn.model_selection import GridSearchCV, train_test_split |
36 |
| -from sklearn.dummy import DummyRegressor, DummyClassifier |
37 |
| -from sklearn.pipeline import make_pipeline |
38 |
| -from sklearn.impute import SimpleImputer |
39 |
| -from sklearn.ensemble import StackingRegressor |
| 45 | +from sklearn.utils._testing import ( |
| 46 | + assert_almost_equal, |
| 47 | + assert_array_almost_equal, |
| 48 | + assert_array_equal, |
| 49 | +) |
40 | 50 |
|
41 | 51 |
|
42 | 52 | def test_multi_target_regression():
|
|
0 commit comments