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?