Process optimization using machine learning