(TensorFlow)ラベルとインデックスをどのようにつなげているのかをご教授ください

Question

例えば「A~J」というラベルの場合はどうするの？のような疑問から生まれた質問です。

from __future__ import absolute_import, division, print_function, unicode_literals
import tensorflow as tf

mnist = tf.keras.datasets.mnist
(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0

model = tf.keras.models.Sequential([
  tf.keras.layers.Flatten(input_shape=(28, 28)),
  tf.keras.layers.Dense(128, activation='relu'),
  tf.keras.layers.Dropout(0.2),
  tf.keras.layers.Dense(10, activation='softmax')
])
model.compile(
  optimizer='adam',
  loss='sparse_categorical_crossentropy',
  metrics=['accuracy']
)

model.fit(x_train, y_train, epochs=5)
predictions = model.predict(x_test)

#ランダムに画像を一枚選んで確認
choice = random.randint(0, len(x_test) - 1)
Answer = np.argmax(predictions[choice])
print("Answer:", Answer)
print(predictions[choice])

実行するとランダムに選ばれた結果の
Answer: 3
[2.1797952e-11 1.3550371e-06 8.9729881e-05 9.9982530e-01 6.0890454e-10
1.8325659e-08 1.1303658e-13 7.3808624e-06 2.1408141e-06 7.4108139e-05]

このような値が表示されたとします。このアンサーが正解である時、
「インデックスの３」が「ラベルの３」に該当しているのはわかりますが、それをどのタイミングで関連させているのでしょうか？

A~Jの場合「インデックスの3」＝「ラベルのD」という関連付けはどのタイミングで行っているのでしょうか？

ご教授よろしくお願いいたします。

Answer 1

AIの推論結果が3だった場合にDと表示させたい、という意味でしたら、
Keras+CNNでCIFAR-10の画像分類
の「index_to_label」関数でやってる内容みたいなことをします

Comments

[vibrato]

ありがとうございます。参考にさせていただきます。