当我使用 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.21 和 ELI5 0.10.1 之间不兼容。
我使用的快速修复是将 sklearn 降级到 0.21.3。更好的解决方案是修复错误并向 eli5 存储库提交 PR。
关于python - eli5 show_prediction 仅显示特征权重而不显示示例预测,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/64695099/