Skip to content

Commit

Permalink
dropping predictions from NAM df
Browse files Browse the repository at this point in the history
  • Loading branch information
AnFreTh committed Aug 25, 2024
1 parent 65fdf3d commit a4aa16e
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions stream_topic/NAM/NAM.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit a4aa16e

Please sign in to comment.