Creating the Classification Model
Learn to create, finalize and save the classification model for natural language processing tasks.
We'll cover the following
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.