State estimation using machine learning