Add and disimbiguate logging from feature constructors.

SkBlaz · Jan 3, 2025 · c5c7ffb · c5c7ffb
1 parent 011f570
commit c5c7ffb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/autoBOTLib/features/features_contextual.py b/autoBOTLib/features/features_contextual.py
@@ -47,6 +47,7 @@ def transform(self, documents):
         """
         :param documents: The input set of documents.
         """
+        logging.info("[Contextual Features] Transforming new documents.")
 
         if not isinstance(documents, list):
 

diff --git a/autoBOTLib/features/features_document_graph.py b/autoBOTLib/features/features_document_graph.py
@@ -146,7 +146,7 @@ def transform(self, new_documents):
             new_documents.values.tolist()
 
         if self.verbose:
-            logging.info("Transforming new documents.")
+            logging.info("[RelationalDocs] Transforming new documents.")
 
         all_embeddings = []
 
@@ -160,7 +160,6 @@ def transform(self, new_documents):
             for k, v in self.core_documents.items():
                 dist = self.jaccard_index(doc_split, v)
                 similarities.append(dist)
-
             similarities = np.array(similarities)
             sorted_dists = np.argsort(similarities)[::-1]
             local_neigh_size = self.neigh_size

diff --git a/autoBOTLib/features/features_topic.py b/autoBOTLib/features/features_topic.py
@@ -71,7 +71,7 @@ def transform(self, new_documents):
             new_documents.values.tolist()
 
         if self.verbose:
-            logging.info("Transforming new documents.")
+            logging.info("[TopicDocs] Transforming new documents.")
 
         new_features = np.zeros((len(new_documents), self.ndim))
         for enx, doc in tqdm.tqdm(enumerate(new_documents),
-Original file line number
+Diff line change
@@ Expand Up / @@ -47,6 +47,7 @@ def transform(self, documents): @@
             """
             :param documents: The input set of documents.
             """
+            logging.info("[Contextual Features] Transforming new documents.")
             if not isinstance(documents, list):
@@ Expand Down @@