Classification and regression overview

Stay organized with collections Save and categorize content based on your preferences.

Binary classification models predict a binary outcome (one of two classes). Use this model type for yes or no questions. For example, you might want to build a binary classification model to predict whether a customer would buy a subscription. Generally, a binary classification problem requires less data than other model types.

Multi-class classification models predict one class from three or more discrete classes. Use this model type for categorization. For example, as a retailer, you might want to build a multi-class classification model to segment customers into different personas.

Regression models predict a continuous value. For example, as a retailer, you might want to build a regression model to predict how much a customer will spend next month.

Workflow for creating a classification or regression model and making predictions

The process for creating a classification or regression model in Vertex AI is as follows:

Steps Description
1. Prepare training data Prepare your training data for model training.
2. Create a dataset Create a new dataset and associate your prepared training data to it.
3. Train a model Train a classification or regression model in Vertex AI using your dataset.
4. Evaluate your model Evaluate your newly trained model for prediction accuracy.
5. View model architecture View the hyperparameter logs of the tuning trials and the hyperparameter logs of the final model.
6. Get predictions from your model

If you want real-time predictions, you can deploy your model and get online predictions.

If you don't need real-time predictions, you can make batch predictions requests directly to your model.