Initial dataset size
Training set
Testing set
20,640
14,448
6192