Change dataset and add train, validation, test

[pdifranc]

Issue #, if available:

Description of changes: feature toggle to show differences between local mode and training on the managed infrastructure

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.