The sample in the docs shows you how to create your custom dataset by extending all of the required functions. Of course you need to adjust the logic to suit your needs (read the CSV files and return the required attribute in the __getitem__) as I suggested in my previous reply