Можно разделить набор данных, используя kfolds с тем же% класса, используя Julia 0.6?

Luísa Rosenstock Völtz спросил: 07 октября 2018 в 10:20 в: split

В Julia 0.6 возможно ли разделить набор данных, используя kfold с одинаковым% классов? Я имею в виду, что если у меня есть три класса: синий, красный и зеленый, мне бы хотелось, чтобы в каждом k-кратном номере было одинаковое количество наблюдений (или очень близких) для каждого класса. Как 20% для синего, 20% для зеленого и 20% для красного.

Я использую

folds = kfolds((Xtrain,Ttrain),k=7)
(xtreino1,ytreino1), (xvali1,yvali1) = folds[1]

, но он не возвращает одинаковое количество наблюдений для каждого класса.

0 ответов