如何在 tensorflow.js 中训练 doc2vec 模型？

Question

我想创建一个应用程序来解决 nlp 问题。我需要在前端部署 doc2vec 模型，我正在寻找 tensorflow.js 框架。我需要在 tensorflow.js 中训练一个 doc2vec 模型，以便将其部署在浏览器上。 是否可以在 tensorflow.js 中训练 doc2vec 模型？

Answer 1

是的，如果您可以在 Python 中将 doc2vec 模型训练为 Keras/TensorFlow 模型，您将能够将其转换为 tensorflow.js 模型。

试试这个教程来训练模型： https://radimrehurek.com/gensim/auto_examples/tutorials/run_doc2vec_lee.html