eli5 show_prediction仅显示特征权重,不显示示例预测

问题描述

使用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])

但是我只能看到这样的特征权重:

enter image description here

我想看到的是这样突出显示的文本:

enter image description here

我该如何实现?

解决方法

我偶然发现了同一问题。造成这种情况的原因似乎是 sckit-learn> 0.21 ELI5 0.10.1 之间的不兼容。

我使用的快速修复程序是将sklearn降级为 0.21.3 。更好的解决方案是修复该错误并将PR提交给eli5存储库。

有关更多信息,请参见https://github.com/TeamHG-Memex/eli5/issues/361