read_for_training.py 文件源码

def read_input_features(l, inp=sys.stdin):
    if isinstance(inp, str):
        with open(inp, 'r') as f:
            return read_input_features(f)

    print("%d samples" % l, file=sys.stderr)
    xs = np.zeros((l, flen), np.int16)
    ys = np.zeros((l, n*n*classes), np.int16)

    i = 0
    for line in inp:
        xs[i, :], ys[i, :] = parse_csv_row_xy(line)
        i += 1
        if i % 10000 == 0:
            print("%d read from disk" % i, file=sys.stderr)
    return xs, ys