add sklearn base estimator