我如何使用＆＃34; metrics.mutual_info＆＃34;在scikit的feature.selection中

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。