Update main.py

main.py

@@ -28,13 +28,6 @@ app.add_middleware(
 
 def train_the_model(data,page):
     if page==2:
-# Function to evaluate the model
-        def evaluate_model(model, X_test, y_test):
-            y_pred = model.predict(X_test)
-            accuracy = accuracy_score(y_test, y_pred)
-            print("Accuracy:", accuracy)
-            print("Classification Report:\n", classification_report(y_test, y_pred))
-            return accuracy
             
         new_data = data
         encoders = load('encoders.joblib')
@@ -54,11 +47,17 @@ def train_the_model(data,page):
                     new_data_filled[col] = encoder.transform(new_data_filled[col])
         X_new = new_data_filled.drop('status.name', axis=1)
         y_new = new_data_filled['status.name']
+
+        X_train, X_test, y_train, y_test = train_test_split(X_new,y_new, test_size=0.2, random_state=42)
+        
         xgb_model.fit(X_new, y_new)
-        dump(xgb_model, 'xgb_model.joblib')
+        dump(xgb_model,'xgb_model.joblib')
+        
         print("Model updated with new data.")
-        updated_model_accuracy = evaluate_model(xgb_model, X_test, y_test)
-        print("Updated model accuracy:", updated_model_accuracy)
+        y_pred = xgb_model.predict(X_test)
+        accuracy = accuracy_score(y_test, y_pred)
+        print("Accuracy:", accuracy)
+        print("Classification Report:\n", classification_report(y_test, y_pred))
         
     else:
         data = data