Clean up final print statements with training requirements

[snova-zoltanc]

Summary

Clean up the final print statements to give information to user about training requirements for their newly prepared dataset.

PR Checklist

• [X] My PR is less than 500 lines of code
• [X] I have added sufficient comment as docstrings in my code
• [X] I have made corresponding changes to the documentation