model_classification function

Building the machine learning model for product classification