From 119763bc49931622baf009805d6d62bd9a4688dc Mon Sep 17 00:00:00 2001 From: adamist521 Date: Mon, 26 Dec 2016 18:11:56 +0900 Subject: [PATCH] cross_validation is included in model_selection module since sklearn 0.18 (#1908) --- demo/gpu_acceleration/bosch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/gpu_acceleration/bosch.py b/demo/gpu_acceleration/bosch.py index f717532cc..2294b24e7 100644 --- a/demo/gpu_acceleration/bosch.py +++ b/demo/gpu_acceleration/bosch.py @@ -3,7 +3,7 @@ import pandas as pd import xgboost as xgb import time import random -from sklearn.cross_validation import StratifiedKFold +from sklearn.model_selection import StratifiedKFold #For sampling rows from input file random_seed = 9