Update some strings

chkoar · Jan 13, 2020 · ab57a7d · ab57a7d
1 parent 8dfafc1
commit ab57a7d
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/vfi/_vfi.py b/vfi/_vfi.py
@@ -121,15 +121,14 @@ def predict(self, X):
         Returns
         -------
         y : ndarray, shape (n_samples,)
-            The label for each sample is the label of the closest sample
-            seen during fit.
+            Predicted target values for X.
         """
 
         check_is_fitted(self, ["classes_"])
         X = check_array(X)
 
-        predicted_probabilitiy = self.predict_proba(X)
-        return self.classes_.take((np.argmax(predicted_probabilitiy, axis=1)), axis=0)
+        predicted_probability = self.predict_proba(X)
+        return self.classes_.take((np.argmax(predicted_probability, axis=1)), axis=0)
 
     def predict_proba(self, X):
         """ Return probability estimates for the test vector X.