fixes regression features in regression/trainer.py