在sklearn RandomForestClassifier中,class_weight = None等同于class_weight =" balanced_subsample"?

时间:2018-03-27 17:51:09

标签: python scikit-learn classification random-forest

documentation中的措辞使其看起来像无和" balanced_subsample"是等价的,但我想确保确实如此。

1 个答案:

答案 0 :(得分:2)

文档清楚地表明它们等效:

class_weight=None - 所有课程都应该有一个重量

class_weight='balanced_subsample' - “平衡”模式使用y的值自动调整输入数据中与类频率成反比的权重n_samples / (n_classes * np.bincount(y))

“balanced_subsample”模式与“balanced”相同,只是根据每棵树生长的bootstrap样本计算权重。