如何通过张量流在X_train，y_train，X_test，y_test中分割图像数据集？

Question

如何将图像数据分为X_train，Y_train，X_test和Y_test？

我正在使用具有tensorflow后端的keras

谢谢。

Answer 1

您不必使用tensorflow或keras来划分数据集。如果您安装了sklearn软件包，则只​​需使用它即可：

from klearn.model_selection import train_test_split
X = ...
Y = ...
x_train, x_test, y_train, y_test = train_test_split(X, Y, test_size=0.2)

您也可以出于相同目的使用numpy：

import numpy
X = ...
Y = ...
test_size = 0.2
train_nsamples = (1-test_size) * len(Y)
x_train, x_test, y_train, y_test = X[:train_nsamples,:], X[train_nsamples:, :], Y[:train_nsamples, ], Y[train_nsamples:,]

祝你好运！