search

calmiLovesAI / TensorFlow2.0_Image_Classification

A TensorFlow_2.0 implementation of AlexNet and VGGNet.
MIT License
39 stars 18 forks source link

Is it possible to add each class test accuracy? #2

Open Chen65010445 opened 3 years ago

Chen65010445 commented 3 years ago

Sometimes I thought only total acc would not satisfy my request, so is it possible to show each class test acc during the evaluation?

boguss1225 commented 2 years ago

Yes see my folked repository. (https://github.com/boguss1225/TensorFlow2.0_Image_Classification) It provides following additional features!

  1. Added 25 models from Keras API
  2. Auto Evaluation + Testing after training
  3. Added more metrics (AUC, Precision, Recall)
  4. Added early stop (patient=5) callback
  5. Added Saving Best model callback
  6. Added History saving function