将gensim doc2vec训练过的模型保存在Google Colab上

时间:2020-05-05 07:52:35

标签: nlp google-colaboratory gensim sentiment-analysis doc2vec

我正在使用gensim doc2vec模型在Google colab存储库GPU运行时上训练文本数据,并希望将训练后的模型保存在test.d2v文件中。以下是代码段

T = [TaggedDocument(doc, [i]) for i, doc in enumerate(data['info'])]
model = Doc2Vec(T,alpha=.025, min_alpha=.025, min_count=1)
model.save('test.d2v')

在colab笔记本中生成以下错误。

** /usr/local/lib/python3.6/dist-packages/smart_open/smart_open_lib.py:253:UserWarning:不建议使用此功能,请改用smart_open.open。有关详细信息,请参见迁移说明:https://github.com/RaRe-Technologies/smart_open/blob/master/README.rst#migrating-to-the-new-open-function '有关详细信息,请参见迁移说明:%s'%_MIGRATION_NOTES_URL

1 个答案:

答案 0 :(得分:0)

警告不一定是错误,尤其是仅提及“弃用”的警告。您的保存文件有效吗?如果是这样,可以忽略此警告。有关更多详细信息,请参见undefined behavior