From 01de675a154d45e17d8a9c674b974f0e2c45d6ab Mon Sep 17 00:00:00 2001 From: annamasce Date: Wed, 25 Sep 2019 19:48:35 +0200 Subject: [PATCH] replaced deprecated cross_validation with model_selection --- rep/metaml/folding.py | 2 +- rep/metaml/gridsearch.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/rep/metaml/folding.py b/rep/metaml/folding.py index c7562f65..acddb9c4 100644 --- a/rep/metaml/folding.py +++ b/rep/metaml/folding.py @@ -10,7 +10,7 @@ from six.moves import zip from sklearn import clone -from sklearn.cross_validation import KFold +from sklearn.model_selection import KFold from sklearn.utils import check_random_state from . import utils from .factory import train_estimator diff --git a/rep/metaml/gridsearch.py b/rep/metaml/gridsearch.py index b5a0bfc9..56c99aba 100644 --- a/rep/metaml/gridsearch.py +++ b/rep/metaml/gridsearch.py @@ -88,7 +88,7 @@ import numpy from sklearn.base import clone -from sklearn.cross_validation import StratifiedKFold, KFold +from sklearn.model_selection import StratifiedKFold, KFold from sklearn.ensemble.forest import RandomForestRegressor from sklearn.utils.random import check_random_state