diff --git a/requirements.txt b/requirements.txt index b948a3f..78b0a69 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,5 @@ nose>=1.1.2 scikit-learn>=0.19 matplotlib>=2.0.0 -numpy>=1.8.0 \ No newline at end of file +numpy>=1.8.0 +joblib>=0.13 \ No newline at end of file diff --git a/stability_selection/stability_selection.py b/stability_selection/stability_selection.py index 1f95af8..d1b8094 100644 --- a/stability_selection/stability_selection.py +++ b/stability_selection/stability_selection.py @@ -23,7 +23,7 @@ import matplotlib.pyplot as plt import numpy as np from sklearn.base import BaseEstimator, TransformerMixin, clone -from sklearn.externals.joblib import Parallel, delayed +from joblib import Parallel, delayed from sklearn.feature_selection import SelectFromModel from sklearn.linear_model import LogisticRegression from sklearn.pipeline import Pipeline