def _generate_model(self, filename, num_rows, cached=False): print("Generating model...") model = ItemSetModel() if(not cached): print("Assembling data...") X, Y = model.get_data_sets(num_rows,cache=True) else: print("Loading cached data...") X, Y = model.get_cached_data(num_rows) model.train(X, Y, train_ratio=0.8) print("Saving model to file...") model.save(filename)