fix: silence scikit-learn 1.0.0 errors

iamDecode · iamDecode · commit 586afdd6aee0 · 2021-12-01T23:14:52.000+01:00
diff --git a/Example.ipynb b/Example.ipynb
@@ -52,9 +52,9 @@
    "source": [
     "# Train classifier\n",
     "clf = RandomForestClassifier(n_estimators=100)\n",
-    "clf.fit(X_train, y_train)\n",
+    "clf.fit(X_train.values, y_train)\n",
     "\n",
-    "clf.score(X_test, y_test)"
+    "clf.score(X_test.values, y_test)"
    ]
   },
   {
diff --git a/cvplot/cvplot.py b/cvplot/cvplot.py
@@ -104,8 +104,8 @@ def kernel(instances, fi):
 
     contributions = [explainer(x) for x in np.asanyarray(X)]
 
-    predictions = model.predict(X)
-    probabilities = model.predict_proba(X)
+    predictions = model.predict(X.values)
+    probabilities = model.predict_proba(X.values)
 
     self.value = {
       'contributions': np.asanyarray(cvlines).tolist(),
diff --git a/cvplot/lime.py b/cvplot/lime.py
@@ -38,7 +38,7 @@ def __init__(self, training_data, scale=True, distance_kernel=None, sample_size=
     if scale:
       #self.scaler = StandardScaler(with_mean=True)
       self.scaler = MinMaxScaler()
-      self.scaler.fit(training_data)
+      self.scaler.fit(self.training_data)
 
       for index in self.categorical_features:
         self.scaler.min_[index] = 0

Original file line number	Diff line number	Diff line change
`@@ -52,9 +52,9 @@`
`52`	`52`	`"source": [`
`53`	`53`	`"# Train classifier\n",`
`54`	`54`	`"clf = RandomForestClassifier(n_estimators=100)\n",`
`55`		`- "clf.fit(X_train, y_train)\n",`
	`55`	`+ "clf.fit(X_train.values, y_train)\n",`
`56`	`56`	`"\n",`
`57`		`- "clf.score(X_test, y_test)"`
	`57`	`+ "clf.score(X_test.values, y_test)"`
`58`	`58`	`]`
`59`	`59`	`},`
`60`	`60`	`{`