Generate the correct `parameters.yml` file directly during training

Training generates a params JSON file in results but it's a YAML file that we need for the predict command