使用预测模型提出建议?

时间:2017-11-09 12:59:56

标签: python scikit-learn random-forest

我使用RandomForestClassifier中的sklearn来创建预测模型。现在我的模型准确度很高。

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

非常感谢任何帮助!

1 个答案:

答案 0 :(得分:-2)

RandomForestClassifier的feature_importances_返回一个数组,其中包含每个要素的数字。数组中的数字越大,该功能的重要性越高。您可以使用

将其与您的功能相匹配
df.columns

获取功能,然后将它们与重要性相匹配。然后,您可以使用循环来创建字典以将要素与数字匹配。然后,您可以找到最高的数字并推荐这些功能。希望这有帮助!