我正在使用以下数据集:
https://www.kaggle.com/abdallahalidev/plantvillage-dataset
我有一个预先训练好的模型。我通过以下方式评估了模型:
from tensorflow.keras.models import load_model
model=load_model('plantvillage.h5')
model.evaluate_generator(test_image_gen)
model.metrics_names
但我不知道如何从预训练模型中获取类名。有什么方法或任何声明可以做到这一点吗?