问题描述
我制作了randomforest分类器模型。 它具有约100个X变量(特征)以对Y(角度)进行分类。 我想权衡一些重要功能。 我该怎么办?
解决方法
如果您以
的形式使用sklearn.ensemble.RandomForestClassifier,则可以使用 class_weight 参数实现此目的。class_weight = {class_label_0: class_weight_0,class_label_1: class_weight_1}
跟随重要特征获得的权重要比不那么重要的特征要高。
您可以在类权重的参数部分的SKLearn documentation中对此进行更详细的检查。