Commit 33078cd9 authored by Anirudha Pimpale's avatar Anirudha Pimpale

commit test left

parent 61431537
testModel.model
testModel.model.trainables.syn1neg.npy
testModel.model.wv.vectors.npy
......@@ -10,7 +10,7 @@ def predict():
tokens = "alzheimer's".split()
new_vector = old_model.infer_vector(tokens)
print(tokens)
# print(old_model.docvecs.most_similar("ball"))
# print(old_model.most_similar([new_vector]))
# print(old_model.docvecs.offset2doctag([new_vector]))
......
......@@ -48,7 +48,7 @@ def train():
model = gensim.models.Doc2Vec(size=300, min_count=0, alpha=0.025, min_alpha=0.025)
model.build_vocab(dataset)
for epoch in range(100):
for epoch in range(10000):
print('iteration'+str(epoch+1))
model.train(dataset, epochs=model.iter, total_examples=model.corpus_count)
model.alpha -= 0.002
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment