From e0a00d80945162ce653926a7617d7fe1ca0627a9 Mon Sep 17 00:00:00 2001
From: benjamin-du <32103306+benjamin-du@users.noreply.github.com>
Date: Fri, 7 May 2021 16:35:30 +0200
Subject: [PATCH] Update tree_categorical_variables.py

---
 .../tree_categorical_variables.py                               | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/BlogPosts/Categorical_variables_in_tree_models/tree_categorical_variables.py b/BlogPosts/Categorical_variables_in_tree_models/tree_categorical_variables.py
index 251d85c..ac5a82e 100644
--- a/BlogPosts/Categorical_variables_in_tree_models/tree_categorical_variables.py
+++ b/BlogPosts/Categorical_variables_in_tree_models/tree_categorical_variables.py
@@ -13,7 +13,7 @@
 import pandas as pd
 from sklearn.linear_model import LogisticRegressionCV
 from sklearn.ensemble import RandomForestClassifier
-from sklearn.cross_validation import StratifiedShuffleSplit
+from sklearn.model_selection import StratifiedShuffleSplit
 from sklearn.preprocessing import StandardScaler
 from sklearn.metrics import roc_auc_score
 from sklearn.tree import DecisionTreeClassifier