to fit with sklearn API
to fit with sklearn API