Spaces:

subbunanepalli
/

LogReg

Sleeping

subbunanepalli commited on Jun 17, 2025

Commit

927c3cd

verified ·

1 Parent(s): 0c59e63

Update train.py

Files changed (1) hide show

train.py CHANGED Viewed

@@ -20,7 +20,7 @@ def train_model():
         df["text_input"] = df.apply(create_text_input, axis=1)
         X = df["text_input"]
-        y = df[["Maker_Action", "Escalation_Level", "Risk_Category", "Risk_Drivers", "Investigation_Outcome", "Alert_Status"]]
         X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

         df["text_input"] = df.apply(create_text_input, axis=1)
         X = df["text_input"]
+        y = df[["Maker_Action", "Escalation_Level", "Risk_Category", "Risk_Drivers", "Investigation_Outcome", "Red_Flag_Reason"]]
         X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)