define custom model name for plots and skip batch creation if no train

Custom model name

• add parameter model_display_name to be able to set the model's name in all plots
• if empty this will be the model class name

Skip batch creation if no train

• move command to create training batches inside of if train
• do not create keras iterator for test set, it is never used