Skip to content

Latest commit

 

History

History
10 lines (7 loc) · 352 Bytes

training.rst

File metadata and controls

10 lines (7 loc) · 352 Bytes

Training

Entry-point for train src/train.py

Arguments description:

  • --model_name: Name of the model to be trained (CatBoost or LightGBM).
  • --data_path: Data directory where all the input (train, val, test) .csv files are stored.
  • --exp_name: Name of the training experiment used for logging.