Add missing information predict method of estimator