Skip to content

Commit

Permalink
Merge pull request #156 from KOSASIH/deepsource-transform-d8fc78e4
Browse files Browse the repository at this point in the history
style: format code with Autopep8, Black, ClangFormat, dotnet-format, Go fmt, Gofumpt, Google Java Format, isort, Ktlint, PHP CS Fixer, Prettier, RuboCop, Ruff Formatter, Rustfmt, Scalafmt, StandardJS, StandardRB, swift-format and Yapf
  • Loading branch information
KOSASIH authored May 12, 2024
2 parents 3947abf + cacfcec commit e66d583
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions ai/risk_assessment/model_training.py
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand All @@ -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.
Expand Down

0 comments on commit e66d583

Please sign in to comment.