Skip to content

Instantly share code, notes, and snippets.

@pbnsilva

pbnsilva/1d_ecg5_dataloader.py

Created April 12, 2020 17:11
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save pbnsilva/2aa5fff6b30127136d7d6ab4e338bbaa to your computer and use it in GitHub Desktop.
Save pbnsilva/2aa5fff6b30127136d7d6ab4e338bbaa to your computer and use it in GitHub Desktop.
Download ZIP
Raw
1d_ecg5_dataloader.py
batch_size = 32
dataset = PhysioNetDataset(ref_file='training2017/REFERENCE.csv', data_dir='training2017')
train_size = int(0.7*len(dataset))
train_set, val_set = random_split(dataset, [train_size, len(dataset)-train_size])
train_loader = DataLoader(train_set, batch_size=batch_size, shuffle=True, num_workers=4)
val_loader = DataLoader(val_set, batch_size=batch_size, shuffle=True, num_workers=4)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment