ge_cmd.py 文件源码

def ge_cmd_learn():
    args = parse_arg_learn()

    # prepare input to GE_learn
    data = GE_data()
    data.dat = util.load_data(args.data)
    data.labeled_features = util.load_labeled_features(args.labeled_features)
    init_model = GE_model()
    param = GE_param()
    if args.l2:
        param.l2_regularization = args.l2
    final_model_path = args.model

    # print data

    final_model = GE_learn(data, init_model, param)
    util.save_model(final_model, final_model_path)
    return

# parse arguments, get data and model, output prediction