Replace fit_transform method with the new fit_sample API.

Regards, Chris
glemaitre · Jun 25, 2016 · bca099d · bca099d
1 parent 20b2458
commit bca099d
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],