我使用RandomForestClassifier
中的sklearn
来创建预测模型。现在我的模型准确度很高。
现在我想建议应该使用哪些输入,以便最大限度地获得我想要的值。我从feature_importances_
获得了权重,但我不知道如何将其转化为推荐。
非常感谢任何帮助!
答案 0 :(得分:-2)
RandomForestClassifier的feature_importances_返回一个数组,其中包含每个要素的数字。数组中的数字越大,该功能的重要性越高。您可以使用
将其与您的功能相匹配df.columns
获取功能,然后将它们与重要性相匹配。然后,您可以使用循环来创建字典以将要素与数字匹配。然后,您可以找到最高的数字并推荐这些功能。希望这有帮助!