Skip to content

Commit

Permalink
Use joblib instead of sklearn.externals.joblib
Browse files Browse the repository at this point in the history
  • Loading branch information
vakker authored and glouppe committed Jan 14, 2020
1 parent 7720fda commit 5160323
Show file tree
Hide file tree
Showing 7 changed files with 9 additions and 9 deletions.
4 changes: 2 additions & 2 deletions benchmarks/bench_ml.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,8 +32,8 @@
from sklearn.datasets import fetch_mldata
from sklearn.datasets import load_boston
from sklearn.datasets import load_digits
from sklearn.externals.joblib import delayed
from sklearn.externals.joblib import Parallel
from joblib import delayed
from joblib import Parallel
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.svm import SVC
Expand Down
2 changes: 1 addition & 1 deletion examples/parallel-optimization.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@
"source": [
"from skopt import Optimizer\n",
"from skopt.space import Real\n",
"from sklearn.externals.joblib import Parallel, delayed\n",
"from joblib import Parallel, delayed\n",
"# example objective taken from skopt\n",
"from skopt.benchmarks import branin\n",
"\n",
Expand Down
2 changes: 1 addition & 1 deletion skopt/learning/gbrt.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from sklearn.base import BaseEstimator, RegressorMixin
from sklearn.ensemble import GradientBoostingRegressor
from sklearn.utils import check_random_state
from sklearn.externals.joblib import Parallel, delayed
from joblib import Parallel, delayed


def _parallel_fit(regressor, X, y):
Expand Down
2 changes: 1 addition & 1 deletion skopt/optimizer/optimizer.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@

from sklearn.base import clone
from sklearn.base import is_regressor
from sklearn.externals.joblib import Parallel, delayed
from joblib import Parallel, delayed
from sklearn.multioutput import MultiOutputRegressor
from sklearn.utils import check_random_state

Expand Down
2 changes: 1 addition & 1 deletion skopt/searchcv.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@

import sklearn
from sklearn.base import is_classifier, clone
from sklearn.externals.joblib import Parallel, delayed
from joblib import Parallel, delayed
from sklearn.model_selection._search import BaseSearchCV
from sklearn.utils import check_random_state
from sklearn.utils.fixes import MaskedArray
Expand Down
2 changes: 1 addition & 1 deletion skopt/tests/test_searchcv.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
from sklearn.tree import DecisionTreeClassifier
from sklearn.base import clone
from sklearn.base import BaseEstimator
from sklearn.externals.joblib import cpu_count
from joblib import cpu_count

from skopt.space import Real, Categorical, Integer
from skopt import BayesSearchCV
Expand Down
4 changes: 2 additions & 2 deletions skopt/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,8 +6,8 @@
from scipy.optimize import minimize as sp_minimize
from sklearn.base import is_regressor
from sklearn.ensemble import GradientBoostingRegressor
from sklearn.externals.joblib import dump as dump_
from sklearn.externals.joblib import load as load_
from joblib import dump as dump_
from joblib import load as load_

from .learning import ExtraTreesRegressor
from .learning import GaussianProcessRegressor
Expand Down

0 comments on commit 5160323

Please sign in to comment.