下記コードはmnistの手書き数字データなので、それぞれの画像1枚につき一つの答え(0~9のうち一つ)を持ちますが、
例えば、動物園で撮った写真に「ぞう」「きりん」「うし」が映っている場合
「ぞう・きりん・うし」という複数解を得たい場合があります。その場合どうすべきでしょうか?
(「しか・らいおん・へび・ぱんだ・うし・くじゃく・きりん」の7つの回答が与えられてるとします。)
ご教授お願いいたします。
python
1from __future__ import absolute_import, division, print_function, unicode_literals 2import tensorflow as tf 3 4mnist = tf.keras.datasets.mnist 5(x_train, y_train), (x_test, y_test) = mnist.load_data() 6x_train, x_test = x_train / 255.0, x_test / 255.0 7 8model = tf.keras.models.Sequential([ 9 tf.keras.layers.Flatten(input_shape=(28, 28)), 10 tf.keras.layers.Dense(128, activation='relu'), 11 tf.keras.layers.Dropout(0.2), 12 tf.keras.layers.Dense(10, activation='softmax')# (7, activation='softmax') 13]) 14model.compile( 15 optimizer='adam', 16 loss='sparse_categorical_crossentropy', 17 metrics=['accuracy'] 18) 19 20model.fit(x_train, y_train, epochs=5) 21 22model.evaluate(x_test, y_test, verbose=2)
回答2件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。