038: Added fasttext experiments

HHansi · Sep 3, 2020 · 465bdd2 · 465bdd2
1 parent fbe1751
commit 465bdd2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/embed2detect/word_embedding_learner.py b/embed2detect/word_embedding_learner.py
@@ -14,7 +14,7 @@ def learn_embeddings(data_folder_path, model_folder_path):
         learn_word2vec_bulk(data_folder_path, model_folder_path, learn_type=learn_type,
                             min_word_count=min_word_count, vector_size=vector_size, window_size=context_size,
                             worker_count=we_workers)
-    if model_type == 'sg':
+    if model_type == 'ft':
         learn_fasttext_bulk(data_folder_path, model_folder_path, learn_type=learn_type, min_word_count=min_word_count,
                             vector_size=vector_size,
                             window_size=context_size, worker_count=we_workers)