Как распечатать метки набора данных тензорного потока? - PullRequest
0 голосов
/ 19 октября 2019

Я хотел бы знать, какие имеются метки в конкретном наборе данных. В коде я знаю метки, но я хочу, чтобы он печатался из набора данных, предполагая, что я не знаю все метки. Есть ли способ сделать это?

Я не мог найти решение для этого в сети.

splits = tfds.Split.ALL.subsplit(weighted=(70,30))

(training_set, validation_set),dataset_info = tfds.load('tf_flowers', with_info = True , as_supervised = True,split = splits)

num_classes = dataset_info.features['label'].num_classes

num_training_examples = 0
num_validation_examples = 0

for example in training_set:
  num_training_examples += 1

for example in validation_set:
  num_validation_examples += 1

print('Total Number of Classes: {}'.format(num_classes))
print('Total Number of Training Images: {}'.format(num_training_examples))
print('Total Number of Validation Images: {} \n'.format(num_validation_examples))

1 Ответ

0 голосов
/ 19 октября 2019

class_names = np.array (dataset_info.features ['label']. Names)

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...