make_data_set.py 文件源码

def make_data_set(rng, samples):
    input_data = []
    output_data = []
    for i in xrange(samples):
        length = rng.randint(100,200,1)
        start_signal = make_start_signal(rng, length)
        duration_signal = make_duration_signal(rng, length)
        target_signal = make_target_signal(start_signal, duration_signal)
        input_data.append(np.concatenate([start_signal.reshape([length[0],1]),duration_signal.reshape([length[0],1])],axis=1))
        output_data.append(target_signal)
    return input_data, output_data


######                Create klepto file
########################################