стратифицированная K-кратная перекрестная проверка по MNIST (классификация по 10 классам) - PullRequest
Новая
       79

стратифицированная K-кратная перекрестная проверка по MNIST (классификация по 10 классам)

0 голосов
/ 05 февраля 2020

Я работаю с данными MNIST и хочу провести стратифицированную проверку Kfold для данных. Я использую CNN (керас) в качестве модели.

Для стратифицированной k-кратной складки я хочу классификацию по меткам (от 0 до 9). Как кодировать многослойный k-кратный код для 10 классов. Какой параметр я должен добавить к коду ?? 

skf = StratifiedKFold(n_splits=K)
skf.get_n_splits(x, y)

for train_index, test_index in skf.split(x, y):
    X_train, X_test = X[train_index], X[test_index]
    y_train, y_test = y[train_index], y[test_index]
    for i, params in enumerate(parameters):
        l = Algorithm(params)
        l.learn(X_train,y_train)
        acc = l.predict(X_test,y_test)
        print(acc)
...