gpleiss / temperature_scaling

A simple way to calibrate your neural network.
MIT License
1.09k stars 159 forks source link

max_iter in set_temperature() #28

Open czb2133 opened 2 years ago

czb2133 commented 2 years ago

I want to adopt temperature_scaling on my own dataset, but I found self.temperature changed just a little from 1.5 to1.495 with lr=0.01 and max_iter=100. But it can continue to decrease with more iterations. So should I increase max_iter?