def main()

in diabetes_regression/training/train.py [0:0]

• 11 lines of code
• 3 McCabe index (conditional complexity)

def main():
    print("Running train.py")

    # Define training parameters
    ridge_args = {"alpha": 0.5}

    # Load the training data as dataframe
    data_dir = "data"
    data_file = os.path.join(data_dir, 'diabetes.csv')
    train_df = pd.read_csv(data_file)

    data = split_data(train_df)

    # Train the model
    model = train_model(data, ridge_args)

    # Log the metrics for the model
    metrics = get_model_metrics(model, data)
    for (k, v) in metrics.items():
        print(f"{k}: {v}")