Skip to content

Commit

Permalink
Issue #2 reproducibility fix
Browse files Browse the repository at this point in the history
  • Loading branch information
jukiewiczm committed Feb 15, 2020
1 parent 8e2a16a commit d68ec2c
Showing 1 changed file with 0 additions and 5 deletions.
5 changes: 0 additions & 5 deletions categorical_variables_embeddings/generate_dense_datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,11 +14,6 @@


class GensimDenseGenerator(DenseDatasetGeneratorParametrized):
def __init__(self, embedding_size, num_epochs, learning_rate):
super(GensimDenseGenerator, self).__init__(embedding_size, num_epochs, learning_rate)
# Ensure reproducibility
os.environ["PYTHONHASHSEED"] = "1245424"

def _get_vectors_container(self, processed_sentences):
model = gensim.models.Word2Vec(processed_sentences,
min_count=1, size=self.embedding_size, batch_words=100, workers=1,
Expand Down

0 comments on commit d68ec2c

Please sign in to comment.