时间戳

首页 > 解决方案 > 使用朴素贝叶斯获取 TF/Count 向量

python - 使用朴素贝叶斯获取 TF/Count 向量

问题描述

我正在使用 Python 3.7（Windows，64 位）。虽然我可以获得 TF-IDF 结果，但之后我无法获得 TF/Count 向量。这是我获得 TF-IDF 结果的示例代码，

DOC, LBL = read_corpus()

# Splits the dataset into training (75%) and test set(25%)
split_point = int(0.75*len(DOC))
trainDoc = DOC[:split_point]
trainClass = LBL[:split_point]
testDoc = DOC[split_point:]
testClass = LBL[split_point:]

# Calling the classifier (use the tf-idf/count feature/vectorizer)
Multinomial_Naive_Bayes(trainDoc, trainClass, testDoc, testClass, tfIdf=True)

标签： pythonnaivebayes

解决方案

做了一些谷歌搜索后，我找不到任何库包含Multinomial_Naive_Bayes()方法。

Multinomial_Naive_Bayes()您可能需要编辑您的问题并在此处粘贴您自己的方法read_corpus()。

但是如果你使用scikit-learn，你可以这样做：

vectorizer = CountVectorizer()
transformer = TfidfTransformer()
word_freq_matrix = vectorizer.fit_transform(corpus)
tfidf_result = transformer.fit_transform(word_freq_matrix)

推荐阅读