Cleaned some code
Closes #14 (closed)
- removed train-test-split in the Dataset classes
- jitted the train step func directly and removed decorator
Fraunhofer GitLab CE and EE, Mattermost, and Harbor Registry will be unreachable due to maintenance on Monday, October 28th, between 5:00 pm CET and 7:00 pm CET. Please log off during the maintenance; otherwise, data loss may occur.
Closes #14 (closed)