diff --git a/gensim/models/doc2vec.py b/gensim/models/doc2vec.py index 08cbf7f106..57693e0eed 100644 --- a/gensim/models/doc2vec.py +++ b/gensim/models/doc2vec.py @@ -454,7 +454,7 @@ def __init__(self, documents=None, input_streams=None, dm_mean=None, dm=1, dbow_ dm : {1,0}, optional Defines the training algorithm. If `dm=1`, 'distributed memory' (PV-DM) is used. Otherwise, `distributed bag of words` (PV-DBOW) is employed. - size : int, optional + vector_size : int, optional Dimensionality of the feature vectors. window : int, optional The maximum distance between the current and predicted word within a sentence. @@ -480,7 +480,7 @@ def __init__(self, documents=None, input_streams=None, dm_mean=None, dm=1, dbow_ useful range is (0, 1e-5). workers : int, optional Use these many worker threads to train the model (=faster training with multicore machines). - iter : int, optional + epochs : int, optional Number of iterations (epochs) over the corpus. hs : {1,0}, optional If 1, hierarchical softmax will be used for model training.