TuneWrapper can now also tune threshold when predict is used