train_insurance_fraud_model module

train_insurance_fraud_model.best_params_from_mlflow() → dict

Get the best parameters from MLflow

train_insurance_fraud_model.insurance_fraud_model() → None

Orchestration flow to train the insurance fraud model and log the metrics and model to MLflow

train_insurance_fraud_model.read_data() → DataFrame

Reads the data from the database

train_insurance_fraud_model.split_data(data: DataFrame, test_size: float = 0.2) → tuple

Split dataset into training and testing sets

train_insurance_fraud_model.train_model(X_train: DataFrame, y_train: Series, X_test, y_test) → None

Train the model and log the metrics and model to MLflow