Before revision:  At the first stage, I estimate the  law of motion of the state variable and the equilibrium policy function using modern machine learning methods.
After revision:  At the first stage, I estimate the state variable's law of motion of the state variable and the equilibrium policy function using modern machine learning methods.