BUGFIX: missing comma when retraining on valid set