python的sklearn extractreesclassifi具有重要功能

2024-09-29 03:23:43 发布

您现在位置:Python中文网/ 问答频道 /正文

尝试应用sklearn ExtraTreesClassifier()来检测数据集(新数据帧)中功能的重要性。因此,我从数据集(x)和列名(y)中选择值,并尝试建立一个模型:

from sklearn.ensemble import ExtraTreesClassifier

arrX=new_y.values
arrY=new_y.columns.values

model = ExtraTreesClassifier()
model.fit(arrX,arrY)

model.feature_importances_

它会给我一个错误,比如“标签数量=18不匹配样本数量=161389” 那么我做错了什么,应该如何正确地做呢?在


Tags: 数据from模型import功能new数量model