예제 #1
0
    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)