我想使用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
我如何将这些整合到我的代码中?
谢谢帮助
回答:
新的scikit-learn版本0.18增加了对互信息特征选择的支持。因此,无需使用metrics.mutual_info
。您可以像使用chi2
一样,在SelectKBest
或SelectPercentile
中使用新的feature_selection.mutual_info_classif评分函数。
X_new = SelectKBest(mutual_info_classif, k=100).fit_transform(X, y)
有关最近更改的更多信息,请查看更新日志。