我使用了sklearn
中的RandomForestClassifier
创建了一个预测模型。现在我有一个精度不错的模型。
现在,我想建议应该使用哪些输入,以最大限度地提高获得我想要的值的机会。我从feature_importances_
那里得到了权重,但我不知道如何将其转化为建议。
任何帮助将不胜感激!
RandomForestClassifier 的feature_importances_返回一个数组,其中包含每个特征的数字。数组中的数字越大,特征的重要性就越高。您可以使用以下命令将其与您的功能相匹配
df.columns
以获取功能,然后将其与重要性相匹配。然后,您可以使用循环遍历并制作字典以将特征与数字匹配。然后,您可以找到最高的数字并推荐这些功能。希望这有帮助!