From 0182c83aeb6608cbea3d05a045cabaf21e5c4637 Mon Sep 17 00:00:00 2001
From: Gautam K <6892333+gautampk@users.noreply.github.com>
Date: Tue, 1 Oct 2019 17:40:03 +0100
Subject: [PATCH] sklearn.cross_validation is depreciated

Add a try/catch to try the new version (sklearn.model_selection) if cross_validation fails to load.
---
 sknn/mlp.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/sknn/mlp.py b/sknn/mlp.py
index c1b2b41..dcf4854 100644
--- a/sknn/mlp.py
+++ b/sknn/mlp.py
@@ -19,7 +19,11 @@
 import sklearn.base
 import sklearn.pipeline
 import sklearn.preprocessing
-import sklearn.cross_validation
+
+try:
+    import sklearn.cross_validation
+except ModuleNotFoundError:
+    import sklearn.model_selection
 
 from .nn import NeuralNetwork, Layer, Convolution, Native, ansi
 from . import backend