F16Sam/awss

1import numpy as np
2from sklearn.metrics import classification_report
3
4def evaluate_model(model, val_generator):
5    val_generator.reset()
6    preds = model.predict(val_generator)
7    y_pred = np.argmax(preds, axis=1)
8    y_true = val_generator.classes
9    labels = list(val_generator.class_indices.keys())
10
11    report = classification_report(y_true, y_pred, target_names=labels, output_dict=True)
12    print(classification_report(y_true, y_pred, target_names=labels))
13
14    return report
15