Function read

artifact/load_data.py:65–69

(data_dir, split)

def read(data_dir, split):
    filename = split + ".pt"
    x, y = torch.load(os.path.join(data_dir, filename))

    return TensorDataset(x, y)

if __name__ == '__main__':
    data = load()