use command line arguments to train model
generator which reads command line arguments
fix generator