- [ ] Create dataloader.py and implement get_data method to get required data - [ ] Change the same throughout the code - [ ] Implement validation dataloader