Machine learning based model predictive control with koopman linearization