ge_cmd.py 文件源码

def ge_cmd_predict():
    args = parse_arg_predict()

    # prepare input to GE_learn
    data = util.load_data(args.data)
    model = util.load_model(args.model)
    pred_path = args.output

    pred = GE_predict(data, model)
    util.write_prediction(pred, pred_path)
    return