Output Threshold at Action Recognition

Hello,

i´m training a custom dataset at I3D_Resnet50_custom Model in Action Recognition.

If im using inferene.py to test my trained model on various Testvideos i´ll get this output:
0000/0010: Video1.MP4 is predicted to class 0
0001/0010: Video2.MP4 is predicted to class 0

Can i output the probability too (e.g. … to class 0 with 56%)?

Because i want to adjust the threshold. My Model predicts everything to my learned 2 classes, even if there´s none of the classes in the video. There it should output “none”.

Greetings and thanks for your help :),

Thomas