How to use and obtain Thresholds to avoid sensitivity in a Multi-class Image Classifier?

Follow the full discussion on Reddit.
I have a Multi-class Image classifier ML model(made in PyTorch) that is integrated into a Flutter app for real-time image classification. Currently, the model is very sensitive and the predicted label quickly switches between the available classes, even with the very small vibrations (or shaking) of the phone camera, caused by holding the phone in a hand by the user. In total there are 39 labels.


There's unfortunately not much to read here yet...

Discover the Best of Machine Learning.

Ever having issues keeping up with everything that's going on in Machine Learning? That's where we help. We're sending out a weekly digest, highlighting the Best of Machine Learning.

Join over 900 Machine Learning Engineers receiving our weekly digest.

Best of Machine LearningBest of Machine Learning

Discover the best guides, books, papers and news in Machine Learning, once per week.