dotnet / dotnet/machinelearning

Cross validation with stratified folds

Open
#4,396 3 comments 0 reactions 0 assignees View on GitHub
area-Core enhancement Priority:2
Dominant language
C#
Stars
9.4k
Forks
2k
Avg merge
2d 20h
Merged PRs (30d)
11

Description

There does not appear to be a way to stratify data in ML.NET, is this likely to be implemented anytime soon?

Say I have data that has an uneven predictor field split 90% / 10%, I would like to cross-validate the data with k folds so that each fold will produce an even predictor split of 50% / 50% (or any desired split setting value).

This does not seem possible yet but is a major feature that is required as part of ML modelling.

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.