问题描述
使用eli5 show预测功能时,我只能看到特征权重,而不能看到突出显示颜色的示例。
我有以下sklearn管道
pipeline = Pipeline([
('tfidf',TfidfVectorizer(analyzer='word',use_idf=True,ngram_range=(1,3),sublinear_tf=True)),('clf',LogisticRegression())
])
pipeline.fit(X_train,y_train)
predicted = pipeline.predict(X_test)
我正在按以下方式使用eli5
eli5.show_prediction(pipeline[1],X[0],vec=pipeline[0])
但是我只能看到这样的特征权重:
我想看到的是这样突出显示的文本:
我该如何实现?
解决方法
我偶然发现了同一问题。造成这种情况的原因似乎是 sckit-learn> 0.21 与 ELI5 0.10.1 之间的不兼容。
我使用的快速修复程序是将sklearn降级为 0.21.3 。更好的解决方案是修复该错误并将PR提交给eli5存储库。