Machine learning for video optimization