如何将二进制 NB 分类器扩展到多标签 SKlearn

Question

我已经使用 sklearn 训练了一个 Nb 分类器，但现在它只返回 1 和 -1。是否有可能使其返回 pos、neg 或 neutral ？这是我的代码：

p = Pipeline([
    ("vectorizer", CountVectorizer()),
    ("selector", SelectPercentile(percentile=20)),
    ("nb", MultinomialNB())
])
p.fit(X, y)
joblib.dump(p, 'skclassif.pkl', compress = 1)
print(p.predict(["I hate this car"]))

Answer 1

它将返回您训练它的类 - 因此，如果您希望三个类通过三个类，则y，仅此而已。