在Wikipedia或Google新闻等大型文章数据集上,哪里可以找到经过预训练的doc2vec模型?

时间:2019-04-19 05:02:38

标签: python nlp gensim word2vec doc2vec

在doc2vec模型上苦苦挣扎的维基百科转储工作的美国人,由于需要进行培训,因此毫无经验地设置服务器,因为本地计算机毫无疑问。除了python 2的过时副本外,我找不到训练有素的模型。

1 个答案:

答案 0 :(得分:1)

我不知道在Wikipedia上有任何公开可用的标准gensim def replay(self, batch_size): minibatch = random.sample(self.memory, batch_size) for state, action, reward, next_state, done in minibatch: target = reward if not done: target = reward + self.gamma * np.amax(self.model.predict(next_state)) target_f = self.model.predict(state) target_f[0][action] = target self.model.fit(state, target_f, epochs=1, verbose=0) if self.epsilon > self.epsilon_min: self.epsilon *= self.epsilon_decay 模型。