=y!d]zddlmZmZddlmZddlmZmZddlZddlm Z ddl Z ddl m Z ddl mZmZmZmZddlmZmZmZdd lmZddlZdd lmZd d lmZmZm Z m!Z!m"Z"m#Z#d d l$m%Z%m&Z&d dl'm(Z(d dl)m*Z*d dl+m,Z,m-Z-ddl.m/Z/ddl0m1Z1GddeeZ2Gdde%e2eZ3Gdde&e2eZ4dS))daal_check_versionsklearn_check_version) BaseEnsemble)ABCMetaabstractmethodN)Number)DataConversionWarning)check_random_statecompute_sample_weight check_array deprecated)check_is_fittedcheck_consistent_length _num_samples)ceil)sparse)_validate_targets _check_X_y _check_array _column_or_1d_check_n_features_convert_to_supported)ClassifierMixinRegressorMixin _get_policy)_check_is_fitted) from_tableto_table)_backend)DecisionTreeClassifiercjeZdZedZ ddZdZdZdZdZ dZ d Z d Z d Z d Zd ZdS) BaseForestc d||_| |_| |_| |_||_||_||_||_||_||_ ||_ ||_ ||_ ||_ | |_| |_||_||_||_||_||_||_||_||_||_dSN) n_estimators bootstrap oob_score random_state warm_start class_weight max_samples criterion max_depthmin_samples_splitmin_samples_leafmin_weight_fraction_leaf max_featuresmax_leaf_nodesmin_impurity_decreasemin_impurity_split ccp_alphamax_bins min_bin_size infer_mode splitter_mode voting_modeerror_metric_modevariable_importance_mode algorithm)selfr'r.r/r0r1r2r3r4r5r6r(r)r*r+r,r7r-r8r9r:r;r<r=r>r?kwargss 6lib/python3.11/site-packages/onedal/ensemble/forest.py__init__zBaseForest.__init__9s:)""($(&""!2 0(@%(,%:""4"  ($*&!2(@%"Fc||St|tr|dkrktds\tdrtjdt |r/t dttj |n|S|dkr/t dttj |S|dkr/t dttj |Stdrdnd }td |d t|tj tjfr|S|d kr t dt||zSd S)Nautoz1.3z1.1z`max_features='auto'` has been deprecated in 1.1 and will be removed in 1.3. To keep the past behaviour, explicitly set `max_features=1.0` or remove this parameter as it is also the default value for RandomForestRegressors and ExtraTreesRegressors.sqrtlog2z"sqrt" or "log2"z"auto", "sqrt" or "log2"z:Invalid value for max_features. Allowed string values are .r) isinstancestrrwarningswarn FutureWarningmaxintnprHrI ValueErrornumbersIntegralinteger)r@r3 n_featuresis_classificationallowed_string_valuess rB_to_absolute_max_featuresz$BaseForest._to_absolute_max_featuresps    lC ( ( 8v%%,U33 F,U33- O * ---%6F3q#bgj&9&9":":!!!;EFv%%1c"'*"5"566777v%%1c"'*"5"566777:O;;%7$6$66 "7377788 8 lW%5rz$B C C   #  q#lZ78899 9qrDc4|dSt|tjrtds5d|cxkr|ks'nd}t |||n+||kr%d}t |||t ||z St|tjrtdrntdrAdt |cxkrdks&nd}t ||n@dt |cxkrdks&nd }t ||t |Sd }t|t|) N?z1.2rGz7`max_samples` must be in range 1 to {} but got value {}z6`max_samples` must be <= n_samples={} but got value {}1.0rz:`max_samples` must be in range (0.0, 1.0] but got value {}z6`max_samples` must be in range (0, 1) but got value {}z7`max_samples` should be int or float, but got type '{}') rLrUrVrrTformatfloatReal TypeErrortype)r@ n_samplesr-msgs rB#_get_observations_per_tree_fractionz.BaseForest._get_observations_per_tree_fractions  2 k7#3 4 4 2(// I[5555I5555SC$SZZ ;%G%GHHH6**RC$SZZ ;%G%GHHHy011 1 k7< 0 0 &$U++ >&u-- >E+..3333!3333VC$SZZ %<%<===4E+..22222222RC$SZZ %<%<===%% %G 4 #4#455666rDc|j\}}||j||j}|||j}t |jr|nd}|js|jtd|js|j rtdt|j tj r|j n#tt|j |z}t|jtj r|jn#tt|j|z}id|jt$jkrdndd|jd |jd |jd |d t/|jd n|jd|jd|jdt|jd|dt|jdn|jd|d|d|jdn|jd|jd|jddt |j|j |j!d}|jr |j"dntG|j"|d<tIdr |j%|d<|S)N)rdr-r]zl`max_sample` cannot be set if `bootstrap=False`. Either switch to `bootstrap=True` or set `max_sample=None`.z6Out of bag estimation only available if bootstrap=Truefptyper`doublemethodr:r<observations_per_tree_fractionimpurity_thresholdrK min_weight_fraction_in_leaf_node#min_impurity_decrease_in_split_node tree_countfeatures_per_nodemax_tree_depthrmin_observations_in_leaf_nodemin_observations_in_split_noder4r8r9memory_saving_modeF)r(r=r> class_count)iPer;)&shaper[r3rYrfr-boolr(rTr)rLr1rUrVrRrr0dtyperSfloat32r?r:r<r`r6r2r5r'r/r4r8r9r=r>classes_lenrr;) r@datardrXrprkrrrs onedal_paramss rB_get_onedal_paramszBaseForest._get_onedal_paramssA $  : ::  z4+ACC*.)Q)QT-=*R*?*?&KO NLL*$)G)G!# '~ $"2">%  ~ 3$. 3233 3&0% &"&" 8D ! !'*)I577(8(8 &'1& '"'" 9D " "'**Y688(9(9 ' rz!9!9ggx dn  $/  4+  -.L  !%.6DRR  ,-J  -.L  D$7$?qqTEX  ! " D-# $ !%% &dn--!%!7(,(E+   .  ! 04 0E113 LLM- ( . / / @-1-?M/ *rDct|jtjr#d|jkst d|jzn+d|jcxkrdksnt d|jzt|jtjr#d|jkst d|jzn+d|jcxkrdksnt d|jzd |jcxkrdksnt d |j4tj d t|jdkrt d |j dkrt d |j ht|j tjst d|j z|j dkr't d |j t|jtjr#d|jkst d|jznt d|jzt|jtjr$d|jkst d|jzdSt d|jz)NrGz:min_samples_leaf must be at least 1 or in (0, 0.5], got %srKg?rz`min_samples_split must be an integer greater than 1 or a float in (0.0, 1.0]; got the integer %sr]z^min_samples_split must be an integer greater than 1 or a float in (0.0, 1.0]; got the float %srz)min_weight_fraction_leaf must in [0, 0.5]zThe min_impurity_split parameter is deprecated. Its default value has changed from 1e-7 to 0 in version 0.23, and it will be removed in 0.25. Use the min_impurity_decrease parameter instead.z5min_impurity_split must be greater than or equal to 0z8min_impurity_decrease must be greater than or equal to 0z1max_leaf_nodes must be integral number but was %rz7max_leaf_nodes {0} must be either None or larger than 1z#max_bins must be at least 2, got %sz+max_bins must be integral number but was %rz'min_bin_size must be at least 1, got %sz/min_bin_size must be integral number but was %r)rLr1rUrVrTr0r2r6rNrOrPr5r4r_r8r9)r@s rB_check_parameterszBaseForest._check_parameterss3 d+W-= > > :--- ":#'#8"9:::. -44444444 ":#'#8"9::: d,g.> ? ? ;... "6$(#9":;;;/ .4444"4444 "4$(#9":;;;D18888S8888HII I  " . MM(  ) ) ) &++ "1222  % * *-.. .   *d173CDD ) '()))"Q&& ()/+*-*-... dmW%5 6 6 3 %% !F#'="1222&"$(M233 3 d')9 : : 7))) !J#'#4"5666*)"$($5677 7rDcjd|_d|_t|d|dS)NT)rOF)copy) class_weight_r|rastyper@yrzs rBrzBaseForest._validate_targets/s7! QT***11%e1DDDrDc|jd}|j}|dkrtdtj|gn||}|}|jd}|dkr.||kr(tdt |d|jd|dkrtj||}nt|trtj |||}nct|dd|d }|j dkrtd |j|fkr)td |j|f|S) NrrGz n_samples=1)rzz.sample_weight and X have incompatible shapes: z vs zT Note: Sparse matrices cannot be indexed w/boolean masks (use `indices=True` in CV).FC) accept_sparse ensure_2drzorderz)Sample weights must be 1D array or scalarz'sample_weight.shape == {}, expected {}!)rxrzrTrSasarrayravelr}onesrLrfullrndimr_)r@Xr sample_weightrdrzsample_weight_counts rB_get_sample_weightzBaseForest._get_sample_weight4sGAJ  >>]++ + &3&;$&2(5UDDD &++-- +1!4 ! # #(;y(H(H*!$M 2 2 2 2AGGG =>> > ! # #GIU;;;MM  v . . MGI}EJJJMM(Ue3M!Q&& !LMMM"yl22 !J"(&)|S) NTcsrrzforce_all_finiterrGr^)rrrzvSome inputs do not have OOB scores. This probably means too few trees were used to compute any reliable OOB estimates.) rrSfloat64r{rrzrrxn_features_in_r n_features_rrrtrainr model _onedal_modelr)rYroob_err_accuracy oob_score_oob_err_decision_functionoob_prediction_ oob_err_r2oob_err_predictionreshapeanyrNrO UserWarning) r@rrrmodulerpolicyparams train_results rB_fitzBaseForest._fit[s qRZ0!8881  " "1ag . .//1mDD gaj$U++ 3#2D !!%A}==3FAq-PP1m((++#v| F<%aM::<<< )/ > % A",\-J"K"KD"Q'1 :(<(<$$#-\-D"E"Ed"K'1 3(5(55>>>@>7>7>7@EEE """H)))###JFFF        rDr$) metaclasscxeZdZ dfd ZdZddZdfd Zdfd ZxZS)RandomForestClassifierdginiNrrGrKrFTFclass_responsesbestweightednonehistc t||||||||| | | | | ||||||||||||d|_dS)Nr'r.r/r0r1r2r3r4r5r6r(r)r*r+r,r7r-r8r9r:r;r<r=r>r?TsuperrCrYr@r'r.r/r0r1r2r3r4r5r6r(r)r*r+r,r7r-r8r9r:r;r<r=r>r?rA __class__s rBrCzRandomForestClassifier.__init__s6 %/-%=%)"71%!%#%!'#/%=3  ! ! !4"&rDcNt||j|\}|_|_|Sr&)rr,rr|rs rBrz(RandomForestClassifier._validate_targetss./@ t %0)0),4 t}rDcR||||tjj|Sr&)rr!decision_forestclassification)r@rrrrs rBfitzRandomForestClassifier.fits,yyA}!1@%II IrDct|tjj|}t j|j| t j dS)Nunsafe)casting) rrr!rrrStaker|rrint64)r@rrpredrs rBpredictzRandomForestClassifier.predictseww8#;#JERRw M JJLL     " "## #rDcht|tjj|Sr&)rrr!rrr@rrrs rB predict_probaz$RandomForestClassifier.predict_probas&ww%%a)A)PRWXXXrD)rrNrrGrKrFNrKNTFNFNrKNrrGrrrrrrNNr&) rrrrCrrrr __classcell__rs@rBrrs!!#$"#*,$ $')$( "!"!-%'#)*0!35&5&5&5&5&5&n   IIII######YYYYYYYYYYrDrcjeZdZ dfd Zdfd Zdfd ZxZS)RandomForestRegressorr squared_errorNrrGrKrFTFrrrrrrc t||||||||| | | | | ||||||||||||d|_dS)NrFrrs rBrCzRandomForestRegressor.__init__s6 %/-%=%)"71%!%#%!'#/%=3  ! ! !4"'rDc|t|dr d||dk<|g}t|||tjj|S)N __array__r]rK)hasattrrrr!r regression)r@rrrrrs rBrzRandomForestRegressor.fit/s_  $}k22 :69 ms23*OMww||Aq-$4?HH HrDct|tjj|Sr&)rrr!rrrrs rBrzRandomForestRegressor.predict7s1ww8#;#FNNTTVVVrD)rrNrrGrKrFNrKNTFNFNrKNrrGrrrrrrrr&)rrrrCrrrrs@rBrrs!*#$"#*,$ $')$( "!"!-%'#)*0!35'5'5'5'5'5'nHHHHHHWWWWWWWWWWrDr)5daal4py.sklearn._utilsrrsklearn.ensemblerabcrrrUrrNsklearn.exceptionsr sklearn.utilsr r r r sklearn.utils.validationrrrmathrnumpyrSscipyrsp datatypesrrrrrr common._mixinrrcommon._policyrcommon._estimator_checksrdatatypes._data_conversionrr onedalr! sklearn.treer"r$rrrrDrBrs"////////))))))''''''''444444 <;;;;;;;((((((777777========//////hhhhhhhhhV QYQYQYQYQY_jGQYQYQYQYhAWAWAWAWAWNJ'AWAWAWAWAWAWrD