From bca099d2bb6eada30945eb1e6e11f2d9041e5744 Mon Sep 17 00:00:00 2001 From: chkoar Date: Sat, 25 Jun 2016 22:20:18 +0300 Subject: [PATCH] Replace fit_transform method with the new fit_sample API. Regards, Chris --- .../under-sampling/plot_repeated_edited_nearest_neighbours.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/under-sampling/plot_repeated_edited_nearest_neighbours.py b/examples/under-sampling/plot_repeated_edited_nearest_neighbours.py index 08fdff199..6ebbf8517 100644 --- 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],