Machine learning high availability