Create CustomDataset

[amillert]

CustomDataset(Dataset) class with the use of the corpus:

• Load data in a correct format,
• Enable iterating over the data + indexing,
• Provide parameters important from the learning perspective.