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

Question

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

标签： tensorflowdeep-learningnlptensorflow.jsdoc2vec

Answer 1

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

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