如何从下面的代码中获取x_test和y_test
test_generator = tf.keras.preprocessing.image_dataset_from_directory(
data_dir2, labels ='inferred', label_mode='categorical',
#validation_split=0.2,
#subset="validation",
#labels='inferred',
seed=123,
image_size=(img_height, img_width),
batch_size=64)
1条答案
按热度按时间siv3szwd1#
根据文档image_dataset_from_directory-function -它可能返回元组
(images, labels)
,因此您可以尝试使用for
-loop来创建包含X
和Y
的列表因为它返回一些
EagerTensor
中的值,所以我使用append()
来移动到普通列表中。如果使用
label_mode='int'
,则会得到Y = [ 1, 0, 2 ]
如果使用
label_mode='categorical'
,则会得到Y = [ [0, 1, 0], [1, 0, 0], [0, 0, 1] ]
如果需要包含类名的字符串(并且使用
label_mode='int'
):