dropping predictions from NAM df

AnFreTh · AnFreTh · commit a4aa16e47cd8 · 2024-08-25T14:27:56.000+02:00
diff --git 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

Original file line number	Diff line number	Diff line change
`@@ -323,6 +323,9 @@ def __init__(`
`323`	`323`	`columns=["text", "tokens"], errors="ignore"`
`324`	`324`	`)`
`325`	`325`
	`326`	`+ if "predictions" in self.structured_data.columns:`
	`327`	`+ self.structured_data = self.structured_data.drop(columns=["predictions"])`
	`328`	`+`
`326`	`329`	`self.target_column = target_column`
`327`	`330`
`328`	`331`	`# Combine topic probabilities with structured data`