Commit 1dd5fc3d authored by Maciej Wójcikowski's avatar Maciej Wójcikowski Committed by GitHub
Browse files

Get rid of scikit-learn deprecation warning

parent 3f2475b7
Loading
Loading
Loading
Loading
+1 −2
Original line number Diff line number Diff line
@@ -9,8 +9,7 @@ from deepchem.models import Model
from deepchem.models.sklearn_models import SklearnModel
from deepchem.utils.save import load_from_disk
from deepchem.utils.save import save_to_disk
from sklearn.cross_validation import train_test_split
from sklearn.grid_search import GridSearchCV
from sklearn.model_selection import train_test_split, GridSearchCV
import tempfile