remove print

aimclub · Jun 5, 2024 · 391d6f6 · 391d6f6
1 parent 8ceb219
commit 391d6f6
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 8 deletions.
diff --git a/autotm/algorithms_for_tuning/genetic_algorithm/ga.py b/autotm/algorithms_for_tuning/genetic_algorithm/ga.py
@@ -612,9 +612,4 @@ def init_individ(self):
         return dict_of_stages
 
     def init_population(self):
-        # if random.random() < self.initial_element_stage_probability:
-        #
-        # for i in range(self.max_stages):
-        #     print()
-
         raise NotImplementedError
diff --git a/autotm/algorithms_for_tuning/genetic_algorithm/selection.py b/autotm/algorithms_for_tuning/genetic_algorithm/selection.py
@@ -7,7 +7,6 @@
 
 
 def yield_matching_pairs(pairs, population):
-    # print('Number of pairs: {}'.format(pairs))
     population.sort(key=operator.attrgetter("fitness_value"))
     population_pairs_pool = []
 

diff --git a/autotm/fitness/tm.py b/autotm/fitness/tm.py
@@ -590,7 +590,6 @@ def _get_avg_coherence_score(self, for_individ_fitness=False):
             self.model, s=self.S, b=self.params.basic_topics
         )
         if for_individ_fitness:
-            # print('COMPONENTS: ', np.mean(list(coherences_main.values())), np.min(list(coherences_main.values())))
             return np.mean(list(coherences_main.values())) + np.min(
                 list(coherences_main.values())
             )

diff --git a/autotm/preprocessing/text_preprocessing.py b/autotm/preprocessing/text_preprocessing.py
@@ -149,7 +149,7 @@ def lemmatize_text(df, **kwargs):
     elif lang == "en":
         df["processed_text"] = df[col_to_process].apply(lemmatize_text_en)
     else:
-        print(f"The language {lang} is not known")
+        logger.error(f"The language {lang} is not known")
         raise NameError
     return df