我使用sklearn来获取tf-idf值,如下所示。

from sklearn.feature_extraction.text import TfidfVectorizer
myvocabulary = ['life', 'learning']
corpus = {1: "The game of life is a game of everlasting learning", 2: "The unexamined life is not worth living", 3: "Never stop learning"}
tfidf = TfidfVectorizer(vocabulary = myvocabulary, ngram_range = (1,3))
tfs = tfidf.fit_transform(corpus.values())

现在,我想在矩阵中查看计算出的tf-idf分数,如下所示。
python - 如何在python中的sklearn中打印TF-IDF分数矩阵-LMLPHP

我试图做到这一点如下。
idf = tfidf.idf_
dic = dict(zip(tfidf.get_feature_names(), idf))
print(dic)

但是,然后我得到的输出如下。
{'life': 1.2876820724517808, 'learning': 1.2876820724517808}

请帮我。

最佳答案

多亏了σηγ,我才能从this question找到答案

feature_names = tfidf.get_feature_names()
corpus_index = [n for n in corpus]
import pandas as pd
df = pd.DataFrame(tfs.T.todense(), index=feature_names, columns=corpus_index)
print(df)

关于python - 如何在python中的sklearn中打印TF-IDF分数矩阵,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/46597476/

10-13 00:26