Using machine learning in workload automation