Блог

Когда я использую функцию прогнозирования показа 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. Лучшим решением было бы исправить ошибку и отправить PR в репозиторий eli5.

Для получения дополнительной информации см. https://github.com/TeamHG-Memex/eli5/issues/361