TensorFlow: Как проверить на 1 заданной c строке данных и обучить остальных?

Question

У меня 11 рядов данных, и моя цель состоит в том, чтобы обучить сеть на 10 и проверить на 1 заданной c строке (не случайно).

Цель состоит в том, чтобы пройти проверку на каждом отдельном ряду во время обучения на других 10, пока у меня не появится прогноз для всех 11 рядов.

x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.1)

Разделение «поезд / тест», как показано выше, не похоже, что оно будет работать случайным образом, есть ли способ точно указать, какие строки будут использоваться для обучения и тестирования?

Answer 1

То, что вы ищете, похоже на k-кратную перекрестную проверку. Это будет использовать каждую строку в качестве проверочного набора, и тренироваться на оставшихся k - 1 рядах и так далее. Я бы предложил использовать встроенный метод sklearn.

from sklearn.model_selection import KFold

n_splits = 11
for train_idx, test_idx in KFold(n_splits).split(x):
    x_train, x_test = x[train_idx], x[test_idx]
    y_train, y_test = y[train_idx], y[test_idx]
    # do your stuff