Creating the Classification Model

Learn to create, finalize and save the classification model for natural language processing tasks.

Creating the model

The create_model() function lets us easily train a classification model on the dataset. In this case, we’ll create a CatBoost model because it generally performs well in classification tasks.

catboost = classification.create_model('catboost')

Run the code above in the Jupyter Notebook by clicking the “Click to launch app!” button.

Get hands-on with 1400+ tech skills courses.