Skip to content

Instantly share code, notes, and snippets.

@raichiii
raichiii / esmultioutput.py
Created February 5, 2026 07:40
MultiOutputRegressor class for early stopping. I have confirmed that it works with `xgboost.XGBRegressor`.
from sklearn.base import _fit_context, is_classifier
from sklearn.utils import Bunch
from sklearn.utils.metadata_routing import _routing_enabled, process_routing
from sklearn.utils.multiclass import check_classification_targets
from sklearn.utils.parallel import Parallel, delayed
from sklearn.utils.validation import _check_method_params, _check_response_method, validate_data
from sklearn.multioutput import _fit_estimator, MultiOutputRegressor
class EarlyStoppableMultiOutputRegressor(MultiOutputRegressor):