Gensim의 doc2vec를 사용하여 대형 코퍼스 용 문서 벡터를 만들었습니다. 이제 큰 DataFrame에 Gensim Doc2vec infer_vector()를 사용하는 방법?
model = gensim.models.doc2vec.Doc2Vec(sentences,size = 10, window = 800, min_count = 1, workers=40, iter=10, dm=0)
sentences=gensim.models.doc2vec.TaggedLineDocument('file.csv')
Eg: model.infer_vector('This is a string')
에 대한 문서 벡터를 생성하는 문서 벡터를 사용하여) (방법을 Gensim의 infer_vector이되어 사용하고 있습니다 infer_vector를 통해 전체 DataFrame을 가져 와서 DataFrame의 각 행에 대한 출력 벡터를 얻으시겠습니까?
'df [ 'col']. apply (model.infer_vector)'? –
또는 아마도 이것은 목록과 함께 작동합니다 :'model.infer_vector (df [ 'col'])' –
고마워, df [ 'col']. apply (model.infer_vector) worked – CMM