dotnet / dotnet/machinelearning
Add a stratified trainer to ML.NET
Open
area-Trainers
enhancement
Priority:2
- Dominant language
- C#
- Stars
- 9.4k
- Forks
- 2k
- Avg merge
- 2d 20h
- Merged PRs (30d)
- 11
Description
Add a trainer that splits the data according to a specified categorical column, and trains a model for each category, as well as a "default" model on the whole dataset. At scoring time, the model corresponding to the example's category is used, unless the category was not present in the training set, in which case the default model is used.
Contributor guide
Assessment
This issue has not been assessed yet.