Using machine learning for predictive maintenance in manufacturing