public void save(String fileName) {
   trainer.saveModel(fileName);
 }