eli5:使用两个标签显示“权重”()

我在努力 eli5 以了解术语对某些类别预测的贡献。

您可以运行此脚本:

import numpy as np
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.linear_model import LogisticRegression
from sklearn.pipeline import Pipeline
from sklearn.datasets import fetch_20newsgroups

#categories = ['alt.atheism', 'soc.religion.christian']
categories = ['alt.atheism', 'soc.religion.christian', 'comp.graphics']

np.random.seed(1)
train = fetch_20newsgroups(subset='train', categories=categories, shuffle=True, random_state=7)
test = fetch_20newsgroups(subset='test', categories=categories, shuffle=True, random_state=7)

bow_model = CountVectorizer(stop_words='english')
clf = LogisticRegression()
pipel = Pipeline([('bow', bow),
                 ('classifier', clf)])

pipel.fit(train.data, train.target)

import eli5
eli5.show_weights(clf, vec=bow, top=20)

问题:

当使用两个标签时,不幸的是,输出仅限于一个表:

categories = ['alt.atheism', 'soc.religion.christian']

但是,当使用三个标签时,它还输出三个表。

categories = ['alt.atheism', 'soc.religion.christian', 'comp.graphics']

在第一个输出中漏掉y=0是软件中的一个bug,还是我漏掉了一个统计点? 我希望第一个案子有两张桌子。