87a1cf1a6a2f927995b534149caf1287e234a775,predict.py,,main,#,34

Before Change

    entries = []
    count = 0
    for i, data_pre in enumerate(dataset):
        entry = {}
        data = [torch.unsqueeze(d, 0) for d in data_pre]  // make batch of size 1
        signal = data[0]
        target = data[1] if (len(data) > 1) else None
        prediction = model.predict(signal)

After Change

    entries = []
    count = 0
    for i, data_pre in enumerate(dataset):
        info_data = dataset.get_information(i)
        entry = {"information": info_data} if isinstance(info_data, str) else info_data
        data = [torch.unsqueeze(d, 0) for d in data_pre]  // make batch of size 1
        signal = data[0]
        target = data[1] if (len(data) > 1) else None
        prediction = model.predict(signal)

Instances