There is a load_best_model argument too, that will automatically load your best model (according to a metric you choose). It’s all in the docs 
3 Likes
There is a load_best_model argument too, that will automatically load your best model (according to a metric you choose). It’s all in the docs