Classification YAML References
- 21 Dec 2022
- 1 Minute to read
- Print
- DarkLight
- PDF
Classification YAML References
- Updated on 21 Dec 2022
- 1 Minute to read
- Print
- DarkLight
- PDF
Article summary
Did you find this summary helpful?
Thank you for your feedback
Classification training jobs use the keys listed in the table below.
| YAML Key | Description |
|---|---|
| dataset | The dataset that the job is dependent on |
| train | Model training hyperparameters |
| model | Classification model and parameters |
| loss | Loss function to use |
| eval | Parameters for evaluation |
Was this article helpful?