python - 如何从 imblearn 中的 RandomUnderSampler 获取样本索引

Question

有谁知道在使用 imblearn 的 RandomUnderSampler 欠采样后是否/如何获得所选样本的索引？曾经有参数“return_indices=True”，现在已从新版本中删除，并且可能被属性“sample_indices_”取代。但是，如果我尝试使用该属性，它就不起作用(请参见下面的代码)。我正在使用 imblearn 版本 0.6.2。

russs = RandomUnderSampler(random_state=0,sampling_strategy={6: 600}).fit(X_train_point,y_train_point)
russs.sample_indices_

AttributeError                            Traceback (most recent call last)
<ipython-input-78-8397ba40f19b> in <module>
      1 russs = RandomUnderSampler(random_state=0,sampling_strategy={6: 600}).fit(X_train_point,y_train_point)
----> 2 russs.sample_indices

AttributeError: 'RandomUnderSampler' object has no attribute 'sample_indices'

Answer 1

我还找到了解决方法。由于欠采样仅基于 y_vector，因此可以添加一个反变量而不是 x-vector/array 并将其编写如下:

counter=range(0,len(y_train_point))
index,y_resampled=RandomUnderSampler(random_state=0,sampling_strategy={6:600}).fit(counter,y_train_point)
X_resampled=X_train_point[index]