Merge pull request scikit-learn-contrib#79 from chkoar/fix-fit_sample…

…-in-examples Replace fit_transform method with the new fit_sample API
glemaitre · Jun 25, 2016 · 52ea6fd · 52ea6fd
2 parents 20b2458 + bca099d
commit 52ea6fd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/under-sampling/plot_repeated_edited_nearest_neighbours.py b/examples/under-sampling/plot_repeated_edited_nearest_neighbours.py
@@ -46,7 +46,7 @@
 # Apply the ENN
 print('ENN')
 enn = EditedNearestNeighbours()
-X_resampled, y_resampled = enn.fit_transform(X, y)
+X_resampled, y_resampled = enn.fit_sample(X, y)
 X_res_vis = pca.transform(X_resampled)
 
 ax2.scatter(X_res_vis[y_resampled == 0, 0], X_res_vis[y_resampled == 0, 1],
@@ -60,7 +60,7 @@
 # Apply the RENN
 print('RENN')
 renn = RepeatedEditedNearestNeighbours()
-X_resampled, y_resampled = renn.fit_transform(X, y)
+X_resampled, y_resampled = renn.fit_sample(X, y)
 X_res_vis = pca.transform(X_resampled)
 
 ax3.scatter(X_res_vis[y_resampled == 0, 0], X_res_vis[y_resampled == 0, 1],