Building a Classification Model with PyCaret

Python

In this example, we're using the PyCaret library to train a classification model. First we initialize the setup for the dataset which in this case is the Titanic dataset. Then we compare different classification models, select the best one before training a final model. Lastly, we make predictions on the test dataset using the trained model.

 1|  from pycaret.classification import *
 2|  
 3|  # Initialize setup
 4|  setup(train, target='Survived')
 5|  
 6|  # Compare different models and choose the best one
 7|  best_model = compare_models()
 8|  
 9|  # Train the chosen model
10|  model = create_model(best_model)
11|  
12|  # Make predictions on test data
13|  predictions = predict_model(model, data=test)
Greg He
Classification | Python | Pycaret
Did you find this snippet useful?

Sign up for free to to add this to your code library

Remove Stop Words from Text in DataFrame Column
Python

Pandas | Nltk | Nlp

147
LightGBM Custom Loss Function
Python

LightGBM | Loss function

131
Tuning XGBoost Hyperparameters with Grid Search
Python

XGBoost | Tuning

125
How to Convert DataFrame Values Into Percentages
Python

Lambda | Percentages | Pipe

118
How to Scale Data Using Standard Scaler But Keep Column Names
Python

Scaler | Standard

89
LightGBM Hyperparameter Tuning with GridSearch
Python

Python | LightGBM | Hyperparameter tuning | Gridsearch

89
Calculating Root Mean Squared Error (RMSE) with Sklearn and Python
Python

Rmse

85
Dynamically Create Columns in Pandas Dataframe
Python

Pandas | Columns

67
How to Train a Catboost Classifier with GridSearch Hyperparameter Tuning
Python

Catboost | Hyperparameter tuning

67
Overwriting a Python Logging File Instead of Appending
Python

Logging

63