I have successfully implemented it on the image dataset but I\'m a little stuck on its application on the numerical dataset.