Skip to content

Commit

Permalink
Address issue scikit-learn-contrib#107 - ADASYN docstring (scikit-lea…
Browse files Browse the repository at this point in the history
  • Loading branch information
glemaitre authored Jul 25, 2016
1 parent 8cdfaad commit cfc73bc
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions imblearn/over_sampling/adasyn.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,17 +65,17 @@ class ADASYN(SamplerMixin):
>>> from collections import Counter
>>> from sklearn.datasets import make_classification
>>> from imblearn.under_sampling import TomekLinks
>>> from imblearn.over_sampling import ADASYN
>>> X, y = make_classification(n_classes=2, class_sep=2, weights=[0.1, 0.9],
... n_informative=3, n_redundant=1, flip_y=0,
... n_features=20, n_clusters_per_class=1,
... n_samples=1000, random_state=10)
>>> print('Original dataset shape {}'.format(Counter(y)))
Original dataset shape Counter({1: 900, 0: 100})
>>> tl = TomekLinks(random_state=42)
>>> X_res, y_res = tl.fit_sample(X, y)
>>> ada = ADASYN(random_state=42)
>>> X_res, y_res = ada.fit_sample(X, y)
>>> print('Resampled dataset shape {}'.format(Counter(y_res)))
Resampled dataset shape Counter({1: 897, 0: 100})
Resampled dataset shape Counter({0: 909, 1: 900})
References
----------
Expand Down

0 comments on commit cfc73bc

Please sign in to comment.