How do I SelectKBest using mutual information from a mixture of discrete and continuous features?

后端 未结 2 1281
青春惊慌失措
青春惊慌失措 2021-01-20 01:11

I am using scikit learn to train a classification model. I have both discrete and continuous features in my training data. I want to do feature selection using maximum mutua

2条回答
  •  不要未来只要你来
    2021-01-20 01:46

    You could also use partials as follows:

    from functools import partial
    
    discrete_mutual_info_classif = partial(mutual_info_classif, iscrete_features=[0, 1, 2])
    SelectKBest(score_func=discrete_mutual_info_classif).fit(x, y)
    

提交回复
热议问题