基于文本分类的单词包替代方法

Question

我试图根据字幕来预测电影的类型。 我使用keras创建了一个模型，对于100个epoc，精度为85％。 这个模型使用单词包作为我的keras图层的输入。

输入将是两个列表[[1,0,0,0,0,0,1,1]，[0,0,1,0]]的数组，第一个列表是实际的BOW对话和第二个列表是一个热门编码的clases（训练4种类型的电影）。

现在我想用Word2Vec而不是BOW进行训练。 是否可能，如果是，如何处理这个问题？

提前致谢！