разделить набор данных на поезд и протестировать с помощью тензорного потока

Я хочу разделить свой полный набор данных (все необработанные данные имеют несколько функций) на обучающие и тестовые наборы. Есть ли другой правильный способ разделить мои данные вместо использования scikit-learn train-test-split? а также мне нужно перетасовать мои данные при разделении. (Если предлагаемый метод основан на тензорном потоке, это тоже лучше.)


person Dale Steyn    schedule 15.11.2020    source источник


Ответы (1)


Попробуйте этот код:

import tensorflow as tf
input = tf.random.uniform([100, 5], 0, 10, dtype=tf.int32)
input = tf.random.shuffle(input)
train_ds = input[:90]
test_ds = input[-10:]
person Andrey    schedule 15.11.2020