ML Katas

Custom Dataset for CSV Data

medium (<1 hr) pytorch dataset dataloader tabular data
this year by E

Write a PyTorch Dataset class that loads data from a CSV file containing tabular data (features + labels). Requirements:

  • Use pandas to read the CSV.
  • Convert features and labels to tensors.
  • Implement __len__ and __getitem__.

Test by iterating over a small CSV dataset.