Documentation Index
Fetch the complete documentation index at: https://kumo.ai/docs/llms.txt
Use this file to discover all available pages before exploring further.
loss: <list[str]> (Optional)
Description
The loss type to use during model optimization depending on the task type. Available Options:| Task Type | Available loss options |
|---|---|
| Binary Classification/Multilabel Classification | binary_cross_entropy (default) focal |
| Multiclass Classification | cross_entropy (default) |
| Regression/Forecasting | mae mse huber (default) multi_quantile |
| Temporal Link Prediction | cross_entropy (default) |
| Static Link Prediction | cross_entropy (default) |
| Multilabel Ranking | cross_entropy (default) |
focal loss uses an alpha value of 0.25 (the weighting factor to balance positive vs. negative examples), and a gamma value of 2.0 (the balance between easy vs. hard examples). You can further customize this in the model plan by replacing the string by a dictionary:
huber loss uses a delta value of 1.0. You can further customize this in the model plan by replacing the string by a dictionary:
multi_quantile for regression or forecasting tasks when you want prediction intervals in addition to the median prediction. It trains multiple quantiles with pinball loss and writes TARGET_PRED together with 27 quantile columns named q_0.005, q_0.01, …, q_0.995:
Supported Task Types
- All