使用预测模型提出建议



我使用了sklearn中的RandomForestClassifier创建了一个预测模型。现在我有一个精度不错的模型。

现在,我想建议应该使用哪些输入,以最大限度地提高获得我想要的值的机会。我从feature_importances_那里得到了权重,但我不知道如何将其转化为建议。

任何帮助将不胜感激!

RandomForestClassifier 的feature_importances_返回一个数组,其中包含每个特征的数字。数组中的数字越大,特征的重要性就越高。您可以使用以下命令将其与您的功能相匹配

df.columns

以获取功能,然后将其与重要性相匹配。然后,您可以使用循环遍历并制作字典以将特征与数字匹配。然后,您可以找到最高的数字并推荐这些功能。希望这有帮助!

相关内容

  • 没有找到相关文章

最新更新