我如何在python randomforest中为变量赋予权重

如何解决我如何在python randomforest中为变量赋予权重

我制作了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中对此进行更详细的检查。