Получить название документа в матрице scikit-learn tf-idf
Я создал матрицу tf-idf, но теперь я хочу получить 2 лучших слова для каждого документа. Я хочу передать идентификатор документа, и он должен дать мне лучшие 2 слова.
Прямо сейчас у меня есть этот пример данных:
from sklearn.feature_extraction.text import TfidfVectorizer
d = {'doc1':"this is the first document",'doc2':"it is a sunny day"} ### corpus
test_v = TfidfVectorizer(min_df=1) ### applied the model
t = test_v.fit_transform(d.values())
feature_names = test_v.get_feature_names() ### list of words/terms
>>> feature_names
['day', 'document', 'first', 'is', 'it', 'sunny', 'the', 'this']
>>> t.toarray()
array([[ 0. , 0.47107781, 0.47107781, 0.33517574, 0. ,
0. , 0.47107781, 0.47107781],
[ 0.53404633, 0. , 0. , 0.37997836, 0.53404633,
0.53404633, 0. , 0. ]])
Я могу получить доступ к матрице, указав номер строки, например.
>>> t[0,1]
0.47107781233161794
Есть ли способ получить доступ к этой матрице по идентификатору документа? В моем случае «doc1» и «doc2».
Спасибо