Custom Dataset Class

Implement a PyTorch Dataset over row records with indexed feature tensors and labels.