Implementation of classification in machine learning