forgot an import

IBCNServices · Aug 28, 2018 · 9c20e9a · 9c20e9a
1 parent 4fe486a
commit 9c20e9a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gendis/experiments/lts_small.py b/gendis/experiments/lts_small.py
@@ -28,6 +28,7 @@
 # Scikit-learn
 from sklearn.metrics import accuracy_score
 from sklearn.linear_model import LogisticRegression
+from sklearn.model_selection import GridSearchCV
 
 def parse_shapelets(shapelets):
     shapelets = shapelets.replace(']', '],')[:-2]