From 12e9a299437c19d941bd8d0d700f88022da74676 Mon Sep 17 00:00:00 2001 From: AFThielmann Date: Sun, 25 Aug 2024 12:34:19 +0200 Subject: [PATCH 1/3] removing print statement --- stream_topic/NAM/NAM.py | 1 - 1 file changed, 1 deletion(-) diff --git a/stream_topic/NAM/NAM.py b/stream_topic/NAM/NAM.py index 89525febc8..b859798511 100644 --- a/stream_topic/NAM/NAM.py +++ b/stream_topic/NAM/NAM.py @@ -520,7 +520,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) From 65fdf3dbfe8aacf0ba45472eba277bf0c1386851 Mon Sep 17 00:00:00 2001 From: AFThielmann Date: Sun, 25 Aug 2024 12:34:47 +0200 Subject: [PATCH 2/3] version increase --- stream_topic/__version__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stream_topic/__version__.py b/stream_topic/__version__.py index ba9f224e26..277ccdab5b 100644 --- a/stream_topic/__version__.py +++ b/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" From a4aa16e47cd8f73e50b5f5a82001482ffedf922a Mon Sep 17 00:00:00 2001 From: AFThielmann Date: Sun, 25 Aug 2024 14:27:56 +0200 Subject: [PATCH 3/3] dropping predictions from NAM df --- stream_topic/NAM/NAM.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/stream_topic/NAM/NAM.py b/stream_topic/NAM/NAM.py index b859798511..92d1b480d6 100644 --- a/stream_topic/NAM/NAM.py +++ b/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