Added epoch_count_in_training_set parameter