是否可以使用gensim doc2vec进行分类

时间:2016-04-04 08:29:36

标签: nlp gensim word2vec

我有一些警告性的训练句子。现在我的目标是预测天气传入句子是否是警告信息。我已经完成Amazon Redshift Column Mapping Options但是根据我的理解,它没有考虑新到达的句子来预测它的正面还是负面。

根据我的经验,我发现gensim.doc2vec中每个句子的输出向量也依赖于其他句子,这意味着我们不能直接使用该模型为新到达的句子生成向量。请有人帮我这个。感谢。

1 个答案:

答案 0 :(得分:4)

生成新向量的一种方法是使用infer_vector()函数,该函数将基于训练模型生成新向量。由于在使用此函数时模型被冻结,因此新向量将基于现有的句子向量,但不会更改它们。