CIS519PG
/

News_Classifier_Demo

Text Classification

news-classification

Inference Endpoints

Model card Files Files and versions Community

tigerlinlxt commited on 22 days ago

Commit

c5afc23

•

1 Parent(s): a232772

Update eval_pipeline.py

Change News Outlet to outlet

Files changed (1) hide show

eval_pipeline.py +1 -1

eval_pipeline.py CHANGED Viewed

@@ -24,7 +24,7 @@ def preprocess_data(df):
     try:
         processed_data = []
         for _, row in df.iterrows():
-            outlet = row["News Outlet"].strip().upper()
             if outlet == "FOX NEWS":
                 outlet = "FOXNEWS"
             elif outlet == "NBC NEWS":

     try:
         processed_data = []
         for _, row in df.iterrows():
+            outlet = row["outlet"].strip().upper()
             if outlet == "FOX NEWS":
                 outlet = "FOXNEWS"
             elif outlet == "NBC NEWS":