Initial dataset size

Training set

Testing set

20,640

14,448

6192