Building a job to be submitted to Spark for training