我如何在 scikit 的 feature.selection 中使用 "metrics.mutual_info"

Question

除了 chi2 等之外，我还想使用其他评分函数，但未在此页上列出。

http://scikit-learn.org/stable/modules/feature_selection.html

http://scikit-learn.org/stable/modules/classes.html

例如metrics.mutual_info和metrics.balanced_accuracy_score

如何将它们集成到我的代码中？

感谢帮助

Answer 1

新的scikit-learn 0.18版本，增加了对互信息特征选择的支持。所以没必要使用metrics.mutual_info。您可以在 SelectKBest 或 SelectPercentile 中使用新的 feature_selection.mutual_info_classif 评分函数，就像使用 chi2 一样。

X_new = SelectKBest(mutual_info_classif, k=100).fit_transform(X, y)

有关最新更改的更多信息，请查看 changelog。