Machine learning for equipment reliability