o \h(@sdZddlmZmZddlZddlmZddlm Z m Z m Z m Z m Z ddlmZmZddlmZdd lmZdd lmZdd lmZ dd d ZdddZGddde e edZddZGddde eedZdS)z)Base class for ensemble-based estimators.)ABCMetaabstractmethodN)effective_n_jobs) BaseEstimatorMetaEstimatorMixinclone is_classifier is_regressor)Bunchcheck_random_state)get_tags)_print_elapsed_time)_routing_enabled)_BaseCompositionc CstsId|vrIz$t|||j|||ddWdW|S1s$wYW|StyH}zdt|vrCtd|jj|d}~wwt|||j||fi|Wd|S1sdwY|S)z7Private function used to fit an estimator within a job. sample_weight)rNz+unexpected keyword argument 'sample_weight'z8Underlying estimator {} does not support sample weights.)rrfit TypeErrorstrformat __class____name__) estimatorXy fit_paramsmessage_clsnamemessageexcrY/home/air/segue/gemini/backup/venv/lib/python3.10/site-packages/sklearn/ensemble/_base.py_fit_single_estimators4       r!cCsht|}i}t|jddD]}|dks|dr%|ttjj||<q|r2|j di|dSdS)aSet fixed random_state parameters for an estimator. Finds all parameters ending ``random_state`` and sets them to integers derived from ``random_state``. Parameters ---------- estimator : estimator supporting get/set_params Estimator with potential randomness managed by random_state parameters. random_state : int, RandomState instance or None, default=None Pseudo-random number generator to control the generation of the random integers. Pass an int for reproducible output across multiple function calls. See :term:`Glossary `. Notes ----- This does not necessarily set *all* ``random_state`` attributes that control an estimator's randomness, only those accessible through ``estimator.get_params()``. ``random_state``s not controlled include those belonging to: * cross-validation splitters * ``scipy.stats`` rvs Tdeep random_state__random_stateNr) r sorted get_paramsendswithrandintnpiinfoint32max set_params)rr$to_setkeyrrr _set_random_states+sr1c@sVeZdZdZe ddedddZdddZdd d Zd d Z ddZ ddZ dS) BaseEnsembleaBase class for all ensemble classes. Warning: This class should not be used directly. Use derived classes instead. Parameters ---------- estimator : object The base estimator from which the ensemble is built. n_estimators : int, default=10 The number of estimators in the ensemble. estimator_params : list of str, default=tuple() The list of attributes to use as parameters when instantiating a new base estimator. If none are given, default parameters are used. Attributes ---------- estimator_ : estimator The base estimator from which the ensemble is grown. estimators_ : list of estimators The collection of fitted base estimators. N ) n_estimatorsestimator_paramscCs||_||_||_dSN)rr4r5)selfrr4r5rrr __init__ls  zBaseEnsemble.__init__cCs |jdur |j|_dS||_dS)zMCheck the base estimator. Sets the `estimator_` attributes. N)r estimator_)r7defaultrrr _validate_estimator}s   z BaseEnsemble._validate_estimatorTcsPtj}|jdifddjD|durt|||r&j||S)zMake and configure a copy of the `estimator_` attribute. Warning: This method should be used to properly instantiate new sub-estimators. csi|]}|t|qSr)getattr).0pr7rr sz0BaseEnsemble._make_estimator..Nr)rr9r.r5r1 estimators_append)r7rBr$rrr?r _make_estimators    zBaseEnsemble._make_estimatorcC t|jS)z0Return the number of estimators in the ensemble.)lenrAr?rrr __len__ zBaseEnsemble.__len__cCs |j|S)z.Return the index'th estimator in the ensemble.)rA)r7indexrrr __getitem__rGzBaseEnsemble.__getitem__cCrD)z0Return iterator over estimators in the ensemble.)iterrAr?rrr __iter__rGzBaseEnsemble.__iter__r6)TN) r __module__ __qualname____doc__rtupler8r;rCrFrIrKrrrr r2Qs   r2) metaclasscCs\tt||}tj|||td}|d||d7<t|}||dg|fS)z;Private function used to partition estimators between jobs.)dtypeNr)minrr*fullintcumsumtolist)r4n_jobsn_estimators_per_jobstartsrrr _partition_estimatorss  r[csZeZdZdZeddZeddZddZfdd Z dfd d Z fd dZ Z S)_BaseHeterogeneousEnsembleaBase class for heterogeneous ensemble of learners. Parameters ---------- estimators : list of (str, estimator) tuples The ensemble of estimators to use in the ensemble. Each element of the list is defined as a tuple of string (i.e. name of the estimator) and an estimator instance. An estimator can be set to `'drop'` using `set_params`. Attributes ---------- estimators_ : list of estimators The elements of the estimators parameter, having been fitted on the training data. If an estimator has been set to `'drop'`, it will not appear in `estimators_`. cCstdit|jS)zDictionary to access any fitted sub-estimators by name. Returns ------- :class:`~sklearn.utils.Bunch` Nr)r dict estimatorsr?rrr named_estimatorssz+_BaseHeterogeneousEnsemble.named_estimatorscCs ||_dSr6r^)r7r^rrr r8rGz#_BaseHeterogeneousEnsemble.__init__cCst|jdkr tdt|j\}}||tdd|D}|s&tdt|r,tnt}|D]}|dkrI||sItd|j j |j ddq0||fS) NrzfInvalid 'estimators' attribute, 'estimators' should be a non-empty list of (string, estimator) tuples.css|]}|dkVqdS)dropNrr=estrrr szB_BaseHeterogeneousEnsemble._validate_estimators..zHAll estimators are dropped. At least one is required to be an estimator.raz The estimator {} should be a {}.) rEr^ ValueErrorzip_validate_namesanyr r rrr)r7namesr^ has_estimatoris_estimator_typercrrr _validate_estimatorss* z/_BaseHeterogeneousEnsemble._validate_estimatorsc stjdi||S)a Set the parameters of an estimator from the ensemble. Valid parameter keys can be listed with `get_params()`. Note that you can directly set the parameters of the estimators contained in `estimators`. Parameters ---------- **params : keyword arguments Specific parameters using e.g. `set_params(parameter_name=new_value)`. In addition, to setting the parameters of the estimator, the individual estimator of the estimators can also be set, or can be removed by setting them to 'drop'. Returns ------- self : object Estimator instance. r^Nr`)super _set_params)r7paramsrrr r.sz%_BaseHeterogeneousEnsemble.set_paramsTcstjd|dS)a< Get the parameters of an estimator from the ensemble. Returns the parameters given in the constructor as well as the estimators contained within the `estimators` parameter. Parameters ---------- deep : bool, default=True Setting it to True gets the various estimators and the parameters of the estimators as well. Returns ------- params : dict Parameter and estimator names mapped to their values or parameter names mapped to their values. r^r")rn _get_params)r7r#rqrr r' sz%_BaseHeterogeneousEnsemble.get_paramscsVt}ztdd|jD|j_tdd|jD|j_W|Sty*Y|Sw)Ncs0|]}|ddkrt|djjndVqdSrRraTN)r input_tags allow_nanrbrrr rd#  z>_BaseHeterogeneousEnsemble.__sklearn_tags__..csrsrt)r rusparserbrrr rd'rw)rn__sklearn_tags__allr^rurvrx Exception)r7tagsrqrr ry s   z+_BaseHeterogeneousEnsemble.__sklearn_tags__)T) rrLrMrNpropertyr_rr8rmr.r'ry __classcell__rrrqr r\s   r\)NNr6)rNabcrrnumpyr*joblibrbaserrrr r utilsr r utils._tagsr utils._user_interfacerutils.metadata_routingrutils.metaestimatorsrr!r1r2r[r\rrrr s$      &T