私は、Gensimのdoc2vecを使用して大きなコーパスのためのドキュメントベクトルを作成しました。大きなDataFrameにGensim Doc2vec infer_vector()を使用するには?
sentences=gensim.models.doc2vec.TaggedLineDocument('file.csv')
model = gensim.models.doc2vec.Doc2Vec(sentences,size = 10, window = 800, min_count = 1, workers=40, iter=10, dm=0)
今、私は別のサンプルコーパス
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