Last active
September 18, 2019 16:18
-
-
Save lgallen/8c2404bf6227c772faef40e423aa5007 to your computer and use it in GitHub Desktop.
Helpful hyper-parameters for training doc2vec
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
#doc2vec parameters | |
vector_size = 300 | |
window_size = 15 | |
min_count = 1 | |
sampling_threshold = 1e-5 | |
negative_size = 5 | |
train_epoch = 100 | |
dm = 0 #0 = dbow; 1 = dmpv | |
worker_count = 1 #number of parallel processes |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
All credit to https://github.com/jhlau/doc2vec