dotnet / dotnet/machinelearning

Add a stratified trainer to ML.NET

Open
#4,019 0 comments 1 reaction 0 assignees View on GitHub
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

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.