Refactor training subcommand

The train subcommand should expect and parse a JSON configuration file and make all the necessary checks.

We need to avoid updating source code to change parameters.

Please update the docs accordingly.