Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
more parameters excluded
  • Loading branch information
markroxor authored Oct 9, 2016
1 parent d5149fd commit c18fbb3
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions gensim/models/doc2vec.py
Original file line number Diff line number Diff line change
Expand Up @@ -529,9 +529,7 @@ def repeat(self, word_count):

class Doc2Vec(Word2Vec):
"""Class for training, using and evaluating neural networks described in http://arxiv.org/pdf/1405.4053v2.pdf"""
def __init__(self, documents=None, size=300, alpha=0.025, min_count=5,
max_vocab_size=None, seed=1, workers=1, min_alpha=0.0001,
dm=1, dbow_words=0, dm_concat=0, dm_tag_count=1,
def __init__(self, documents=None, dm=1, dbow_words=0, dm_concat=0, dm_tag_count=1,
docvecs=None, docvecs_mapfile=None, comment=None, trim_rule=None, **kwargs):
"""
Initialize the model from an iterable of `documents`. Each document is a
Expand Down Expand Up @@ -601,8 +599,6 @@ def __init__(self, documents=None, size=300, alpha=0.025, min_count=5,
"""
super(Doc2Vec, self).__init__(
size=size, alpha=alpha, min_count=min_count, max_vocab_size=max_vocab_size,
seed=seed, workers=workers, min_alpha=min_alpha,
sg=(1+dm) % 2,
null_word=dm_concat, **kwargs)
self.dbow_words = dbow_words
Expand Down

0 comments on commit c18fbb3

Please sign in to comment.