0

我正在通过管道预处理数据,将分类转换为数字,编码等......而且非常舒服。

但是在项目后期有一个例子,我想测试一些特征的重要性,我需要给模型 X 和 y。但它不接受管道,因此 X 和 y 没有被预处理。

from yellowbrick.model_selection import FeatureImportances

model = RandomForestClassifier(n_estimators=10)
viz = FeatureImportances(model)
viz.fit(X, y)
viz.show()

有没有办法使用管道预处理数据,如 X,y 输入模型?或者我应该为这种情况手动预处理和编码数据?谢谢

4

1 回答 1

0

设法找到解决方案。

基本上,您可以通过管道 .steps[1][1] 访问具有特征重要性的模型

尽管使用 Ordinal/Label 编码器似乎比使用 One hot 编码器更容易使用此 Yellowbrick 功能重要性。

于 2020-02-06T13:02:47.043 回答