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

eli5 show_prediction shows only feature weights and not example prediction

当我使用 eli5 显示预测功能时,我只看到特征权重,而不是颜色突出显示的示例。

我有以下 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.21ELI5 0.10.1.

之间不兼容

我使用的quickfix是将sklearn降级到0.21.3。更好的解决方案是修复错误并向 eli5 存储库提交 PR。

有关详细信息,请参阅 https://github.com/TeamHG-Memex/eli5/issues/361