我如何在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中对此进行更详细的检查。