pycaret_classification_quickstart_diabetes_train_evaluate_save.py

1# Loading the dataset
2from pycaret.datasets import get_data
3data = get_data('diabetes')
4
5# Initialize setup
6from pycaret.classification import *
7s = setup(data, target = 'Class variable', session_id = 123)
8
9# Train a model
10best = compare_models()
11
12# Analyze model
13evaluate_model(best)
14
15# Predictions on hold-out set
16predictions = predict_model(best)
17
18# Save the model
19save_model(best, 'my_first_pipeline')