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