从 TF-IDF 获取单个句子

Question

所以我在一堆句子上使用了 tf-idf，但现在我想获取单个句子的 TF-IDF 向量。这是我的代码。我现在如何获取文本矩阵中的第一项？

tfidf = TfidfVectorizer(
min_df = 5,
max_df = 0.95,
max_features = 8000
)

tfidf.fit(df_m_a['text_final'])
text = tfidf.transform(df_m_a['text_final'])

Answer 1

您已经与 tfidf.fit(df_m_a['text_final']) 通话过。因此变量 tfidf 包含 TF-IDF 向量化器，代码的最后一部分 text = tfidf.transform(df_m_a['text_final']) 你将得到转换后的向量，你应该用 text[0][=14 得到你想要的=]

从 TF-IDF 获取单个句子

Get individual sentences from TF-IDF

python

tf-idf