diff --git a/ai/risk_assessment/model_training.py b/ai/risk_assessment/model_training.py index 0cb70c619..1b16b7a05 100644 --- a/ai/risk_assessment/model_training.py +++ b/ai/risk_assessment/model_training.py @@ -2,6 +2,7 @@ from sklearn.ensemble import RandomForestClassifier from sklearn.metrics import roc_auc_score + def train_model(X_train, y_train): """ Trains the machine learning model using the training data. @@ -10,6 +11,7 @@ def train_model(X_train, y_train): model.fit(X_train, y_train) return model + def evaluate_model(model, X_test, y_test): """ Evaluates the performance of the trained model using the testing data.