cddf39be59c3c1231d2d4fd3b9bfc10a21e63780,imblearn/under_sampling/prototype_selection/tomek_links.py,TomekLinks,_sample,#TomekLinks#Any#Any#,141

Before Change

        links = self.is_tomek(y, nns, self.ratio_)

        if self.return_indices:
            return (X[np.logical_not(links)], y[np.logical_not(links)],
                    np.flatnonzero(np.logical_not(links)))
        else:
            return X[np.logical_not(links)], y[np.logical_not(links)]

After Change

        idx_under = np.flatnonzero(np.logical_not(links))

        if self.return_indices:
            return (safe_indexing(X, idx_under),
                    safe_indexing(y, idx_under),
                    idx_under)
        else:
            return (safe_indexing(X, idx_under),
                    safe_indexing(y, idx_under))

Instances