diff --git a/transaction_management/model_training.py b/transaction_management/model_training.py index 88cecc188..93f45448d 100644 --- a/transaction_management/model_training.py +++ b/transaction_management/model_training.py @@ -2,6 +2,7 @@ from sklearn.ensemble import RandomForestClassifier from sklearn.metrics import roc_auc_score + def train_transaction_model(X_train, y_train): """ Trains the transaction machine learning model using the training data. @@ -10,6 +11,7 @@ def train_transaction_model(X_train, y_train): model.fit(X_train, y_train) return model + def evaluate_transaction_model(model, X_test, y_test): """ Evaluates the performance of the trained transaction machine learning model using the testing data.