Skip to content

Commit 52ea6fd

Browse files
authored
Merge pull request #79 from chkoar/fix-fit_sample-in-examples
Replace fit_transform method with the new fit_sample API
2 parents 20b2458 + bca099d commit 52ea6fd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/under-sampling/plot_repeated_edited_nearest_neighbours.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@
4646
# Apply the ENN
4747
print('ENN')
4848
enn = EditedNearestNeighbours()
49-
X_resampled, y_resampled = enn.fit_transform(X, y)
49+
X_resampled, y_resampled = enn.fit_sample(X, y)
5050
X_res_vis = pca.transform(X_resampled)
5151

5252
ax2.scatter(X_res_vis[y_resampled == 0, 0], X_res_vis[y_resampled == 0, 1],
@@ -60,7 +60,7 @@
6060
# Apply the RENN
6161
print('RENN')
6262
renn = RepeatedEditedNearestNeighbours()
63-
X_resampled, y_resampled = renn.fit_transform(X, y)
63+
X_resampled, y_resampled = renn.fit_sample(X, y)
6464
X_res_vis = pca.transform(X_resampled)
6565

6666
ax3.scatter(X_res_vis[y_resampled == 0, 0], X_res_vis[y_resampled == 0, 1],

0 commit comments

Comments
 (0)