Как создать набор данных в PyTorch для NN?

Question

Я только начал изучать PyTorch и не совсем понимаю, как создаются классы для обучающих данных. В этой задаче используются готовые данные о молекулах https://colab.research.google.com/drive/1I8a0DfQ3fI7Njc62__mVXUlcAleUclnb?usp=sharing . Я хочу создать свой класс с моими данными, подобными torch_geometric.datasets.TUDataset.