Получить метки значения тестируемого объекта (model.evaluate) нейронной сети

Question

Я пытаюсь обучить нейронную сеть, и я хотел бы знать, как я могу получить значения метки, рассчитанные нейронной сетью, когда я вызываю функцию оценивания.

У меня есть поиск в документации по keras для параметра, который делает это, но я ничего не нахожу.

import tensorflow as tf
from tensorflow import keras
import numpy as np

# Create the array of data

train_data = [[1.0,2.0,3.0],[4.0,5.0,6.0]]
train_data_np = np.asarray(train_data)
train_label = [[1,2,3],[4,5,6]]
train_label_np = np.asarray(train_data)

### Build the model

model = keras.Sequential([
    keras.layers.Dense(3,input_shape =(3,2)),
    keras.layers.Dense(3,activation=tf.nn.sigmoid)
])

model.compile(optimizer='sgd',loss='sparse_categorical_crossentropy',metrics=['accuracy'])

#Train the model

model.fit(train_data_np,train_label_np,epochs=10)

#test the model
restest = model.evaluate(test_data_np,test_label_np)

Answer 1

перед оценкой вашей модели вы должны предсказать метки вашего тестового набора:

#predicting 
predict_labels = model.predict(test_data_np)
#evaluate
restest = model.evaluate(test_label_np,predict_labels)