Import libraries and set style
s2 · run_code workaroundHandle both old and new scikit-learn locations for train_test_split, use seaborn's set_context for the poster style, and remove the Jupyter magic command.
@@ -2,8 +2,11 @@
import itertools
import matplotlib.pyplot as plt
+import seaborn as sns
from sklearn import svm, datasets
-from sklearn.cross_validation import train_test_split
+try:
+ from sklearn.cross_validation import train_test_split
+except ImportError:
+ from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
-from sklearn.metrics import classification_report,confusion_matrix
-plt.style.use('seaborn-poster')
-%matplotlib inline
+from sklearn.metrics import classification_report, confusion_matrix
+sns.set_context('poster')
