Automated classification using machine learning