tf.keras.utils.split_dataset
Разделяет набор данных на левую и правую половины (например, на обучающую и тестовую выборки).
tf.keras.utils.split_dataset(
dataset, left_size=None, right_size=None, shuffle=False, seed=None
)
| Аргументы | |
|---|---|
dataset | A tf.data.Dataset, a torch.utils.data.Dataset object, or a list/tuple of arrays with the same length. |
left_size | Если число с плавающей точкой (в диапазоне [0, 1]), это означает долю данных для включения в левый набор данных. Если целое число, это означает количество образцов для включения в левый набор данных. Если None, по умолчанию это дополнение к right_size. По умолчанию None. |
right_size | Если число с плавающей точкой (в диапазоне [0, 1]), это означает долю данных для включения в правый набор данных. Если целое число, это означает количество образцов для включения в правый набор данных. Если None, по умолчанию это дополнение к left_size. По умолчанию None. |
shuffle | Булево значение, указывает, следует ли перемешать данные перед разделением. |
seed | Случайное семя для перемешивания. |
| Возвращаемые значения | |
|---|---|
Кортеж из двух tf.data.Dataset объектов: левого и правого разделов. |
Пример:
data = np.random.random(size=(1000, 4)) left_ds, right_ds = keras.utils.split_dataset(data, left_size=0.8) int(left_ds.cardinality()) 800 int(right_ds.cardinality()) 200
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/utils/split_dataset