【发布时间】:2021-02-03 21:40:11
【问题描述】:
我正在使用Scikit-learn 来训练分类模型。我的训练数据中既有离散特征又有连续特征。
我想使用mutual information 进行特征选择。
特征 1,2 和 3 是离散的。为此,我尝试了以下代码:
mutual_info_classif(x, y, discrete_features=[1, 2, 3])
但它不起作用,它给了我错误:
ValueError: could not convert string to float: 'INT'
【问题讨论】:
-
我已经应用了 W.P. 先生的代码。 McNeill 曾在stackoverflow.com/q/43643278 求婚,但没有成功
-
我们需要更多信息才能为您提供帮助。如果您复制代码的简化示例,这可能会很有用。
-
这是我的代码:from sklearn.feature_selection importmutual_info_classif res_M_train =mutual_info_classif(data_train, Y_train,discrete_features= [1,2,3]) 谢谢
-
我的数据就是这样:[0.983874,TCP,HTTP,FIN,10,8,816,1172,172,17.278635,62,62,2552,5976.375,375,8342.5342.53125,25,125,25,125,25,19319319331931931931931931931931931931931931931931931931931931931931931931931931931931931931931931931931931931931931931931931999999999999太平洋合物,1624757001,255,0.206572,0.108393,0.098179,82,147,1,184,2,1,1,1,1,2,0,0,1,1,3,0,] 如您所见,我的三个首要特征是分类,并且我想计算每个特征的互信息: from sklearn.feature_selection importmutual_info_classif res_M_train =mutual_info_classif(data_train, Y_train,discrete_features= [1,2,3])
标签: python scikit-learn feature-detection