make_data_set.py 文件源码

def make_target_signal(start_signal, duration_signal):
    target_signal = np.zeros([start_signal.shape[0], 2])
    counter = 0
    for i in xrange(target_signal.shape[0]):
        if start_signal[i] == 1:
            counter = duration_signal[i]
        if counter > 0:
            target_signal[i, 0] = 1
            counter -= 1
    target_signal[:,1] = 1 - target_signal[:,0]
    return target_signal


######                   Create data set
########################################