Merge pull request #93 from AnFreTh/main

v0.1.8

Author: AnFreTh

stream_topic/NAM/NAM.py

@@ -323,6 +323,9 @@ def __init__(
             columns=["text", "tokens"], errors="ignore"
         )
 
+        if "predictions" in self.structured_data.columns:
+            self.structured_data = self.structured_data.drop(columns=["predictions"])
+
         self.target_column = target_column
 
         # Combine topic probabilities with structured data
@@ -520,7 +523,6 @@ def forward(self, x):
 
     def training_step(self, batch, batch_idx):
         x, y = batch
-        print(y)
         y_hat = self(x)
         loss = self.loss_fn(y_hat, y)
 

stream_topic/__version__.py

@@ -1,4 +1,4 @@
 """Version information."""
 
 # The following line *must* be the last in the module, exactly as formatted:
-__version__ = "0.1.7"
+__version__ = "0.1.8"