fake_data_generator.py 文件源码

def generate_ds2_data(max_length, str_w, nout, nbands, batch_size, num_iter):
    frame_stride = 0.01  # seconds, hard-coded value in make_aeon_dataloaders
    max_utt_len = ((int(max_length / frame_stride) - 1) // str_w) + 1
    max_lbl_len = (max_utt_len - 1) // 2

    train_set, eval_set = make_fake_dataloader(nbands, max_lbl_len, max_utt_len,
                                               nout, batch_size, num_iter)

    inputs = train_set.make_placeholders()

    if "audio_length" not in inputs:
        inputs["audio_length"] = ng.placeholder([ax.N], dtype=np.uint32)
    if "trans_length" not in inputs:
        inputs["trans_length"] = ng.placeholder([ax.N], dtype=np.uint32)

    return inputs, train_set, eval_set