runner.py 文件源码

def train_model(parameters):
    X, Y, meta = utils.load_dataset(parameters['prefix'])
    print(X.shape)
    print(Y.shape)
    model_test = model.TournamentNn(input_dim=X.shape[1], labels_dim=Y.shape[1], prefix=parameters['prefix'])
    model_test.train(X, Y, prefix=parameters['prefix'], batch=True)
    # test_query = model_test.predict(query_input)