使用 _BatchDataset 评估指标

问题描述 投票:0回答:0

我正在尝试使用 Keras 库收集我的 CNN 模型的指标。我的测试数据集目前是一个 _BatchDataset(方法)。取消批处理只会创建另一个方法 (UnbatchDataset)。我如何将这些数据转换成可用的形式以使用准确度、f1-score、PrecisionatRecall 和 ROC?

预测

#prediction
y_predict_face = face_model.predict(facial_testing_set)

测试集

facial_testing_set = tf.keras.utils.image_dataset_from_directory(path_1, labels = 'inferred', label_mode= 'categorical', image_size=(48, 48), batch_size= 32, color_mode= 'grayscale')

精度指标

# Evaluate by means of accuracy
accuracy = keras.metrics.categorical_accuracy(facial_testing_set.unbatch, y_predict_face)
print(f'Face Model Accuracy: {accuracy}')
tensorflow machine-learning keras conv-neural-network metrics
© www.soinside.com 2019 - 2024. All rights reserved.