Add Relabel function
Change traing & testing data size
Slight change the architecture to fit the 5 classes output