Train a model

Note

Training on GPU requires NVIDIA Driver of version 418.xx or higher.

Execution format

catboost fit -f <file path> [optional parameters]

Options

See all options in the Training parameters section.

Usage examples

Train a model with 100 trees on a comma-separated pool with header:

catboost fit --learn-set train.csv --test-set test.csv --column-description train.cd  --loss-function RMSE --iterations 100 --delimiter=',' --has-header

Train a classification model on GPU:

catboost fit --learn-set ../pytest/data/adult/train_small --column-description ../pytest/data/adult/train.cd --task-type GPU