.run_regs;!RST 1Ax($S) ""$^4V.run_clf5sK1STU 1Ax($S) ""$^4Y?B#qqUV  1Ax($S) ""$^4Y?B::lL99   r)r6n_labels n_classesr7)sklearn.datasetsrrrr,ndarray)rrrrr3rrr>s` rcheck_init_estimationrDs !2:: !"** ! ! T ;DAq AqM TQR HDAq AqM !2:: !"** ! ! B ?DAq AqM )ab DAq AqMrweightedc  ddlm}ddlm}ddlm}d}d}d}d}t jjd} |||| \} } |r| j| } nd } tj| | | } t jd dg}i}tjd|||d| || dfg|}|j| d}tj|ddsJ|ddddksJt!|j"Dcgc]#}|t%t'|| |||%}}t j(|j*}t!|j*dD]}||}tjd|||d| || dfg|||}tj|ddsJ|ddddksJt j,j/t j|ddt j|dddd|j| |d d |f<t!|j*dD]0}t j,j/|d d |f|d d |f2y cc}w)zTest for quantile loss.r)r)mean_pinball_loss)_metric_decoratorr4gi)r6 n_featuresr7)sizeNweightg??zreg:quantileerror) objectiverquantile_alphar Trainnum_boost_roundevalsr+T) strict_shapequantileg4@) sample_weightalpha)rSrT custom_metricr+g>@rGgư>)atolrtol)rBrsklearn.metricsrGxgboost.sklearnrHr,random RandomStater(QuantileDMatrixarraytrainpredicttmnon_increasingrangerKrremptyshapetestingassert_allclose)rrErrGrHr6rJr!r rngrrrMXyrYr+ booster_multi predt_multiimetricspredtsa booster_is rcheck_quantile_lossruNs011IJLJ ))   %C  DAq +  Q& 1B HHc3Z E$&LII,&#$  $G}o! M '''>K  \'2:> ??   ,R 04 77uzz"   )uQxP!  GXXk'' (F 5;;q> "- !HII0*"#(   (=/!!*%    g!6z!BCCG$Z04t;; "" HH\'*:6 7 HH\'*+>? @ # !((,q!t 1-45;;q> "D ""6!Q$<QT1BCDKs (J n_entriesindptrdatadtypescddlm}|jd|dk(sJ|jd|k(sJ|jtj k(sJt d|jD]g}t||dz }t||}t |dz|D]4}||||dz kDsJ|||dz s"||||dz dzk(r4Jiy)zCheck the cut values.ris_categorical_dtyperWrN) pandas.api.typesr|ridtyper,uint64rgrKint) rvrwrxryr|rpbegendjs r check_cutrs6 ::a=F2J && ::a=I %% <<299 $$ 1fkk "2&Q- &)nsQw$ 2A7T!a%[( (#F1q5M2Aw$q1u+/11 22ruse_cupycddlm}d}d}d}tjg|z}t j |||\}}} t j||| |} | j\} } t|dz|z| | |t j||| } t j||d | | j\} } t|dz|z| | |d } d }t jt j||| |d d i}t j|}t j||d | | j\} } t|dz|z| | |d}t j|||dd\}}|r*ddl}ddl}|j#|}|j%|}t j|||d}|j\} } t||z| | |j&t j||d}t j||d ||j\} } t||z| | |j&t j|||ddd\}}t)|j&Dcgc] }||s dc}}||z }||z|dz|zz}t j|||d}|j\} } t|| | |j&t j||d}t j||d ||j\} } t|| | |j&ycc}w)zCheck with optional cupy.rr{i)r)rMmax_binrrL)rrr8cache Fg?)sparsityNT)renable_categoricalrrN)r cat_ratio)r}r|r,float32rerr(raget_quantile_cutrDMatrixrcIteratorForTest make_batchesmake_categoricalcudfcupy from_pandasrbrylen)rrr|r6rJrryrrwXywrwrx n_batchesn_samples_per_batchitrm n_categoriesrcpr~n_cat_featuresn_num_featuresrvs rcheck_get_quantile_cut_devicers5IJGjj\J &F  JJGAq!**1a7KC'')LFD w{j(&$? ++a1 %CIIkg>D'')LFD w{j(&$?I   ,j)X N  Bkk"oBIIkg>D'')LFD w{j(&$?L  y*lETW XDAq   Q  HHQK  Q7t LB&&(LFD lZ'qxx@ Qd 3BIIkg>C&&(LFD lZ'qxx@   :|USC DAqQ5I%5P!QRN.0N~-10NNI  Q7t LB&&(LFD iqxx0 Qd 3BIIkg>C&&(LFD iqxx0Rs ; M McD|dk(}t|d|r t|dyy)zCheck the quantile cut getter.gpu_histFTN)r)rrs rcheck_get_quantile_cutrs*j(H!+u5%k48rrrowscolsroundscatsdevicectj|||d\}}tj|||d\}} i} i} |t|d} tj||d} tj | | || dfg| tj||d} tj | | || dfg| t jjt j| ddt j| ddd tj| ddsJi}t| d <d | d <tj||d} tj | | || dfg|| dd}|dd}t||D] \}}||k\r Jd | d <i}tj | | d| dfg|tj|ddsJ|y)z0Test for one-hot encoding with categorical data.TF)rmax_cat_to_onehotrrrQrRr%gMbP?r\rr reg_lambdag?rN) rer USE_ONEHOTr(rrcr,rjrkrbrfUSE_PARTzip)rrrrrronehotlabelcat_by_etl_resultsby_builtin_results parametersm by_groupingrmse_oh rmse_grouprsbs rcheck_categorical_oher s  ''dD$?MFE  tT5 9FC8:N<>#' "J FEer) r(rcrerfrdroot_mean_squarer,rjrk)rr+boostery_predtr%rmrrs rrunz&check_categorical_missing..runbs*; &'(* ))  =/%    g!6v!>??//"%""5'2 ""4g)>v)Fr)JQU"Vr)rerr(rrrr) rrrrrrrrmrrs @@@rcheck_categorical_missingrXsj2=!OJ$$ DJC S%D 9BWsWtW& OMrparamdmat num_roundsc"i}tj||||dfgd|}|j|jk(sJ|j |k(sJ|j |j k(sJ|j |j k(sJ|S)z-Get training result from parameters and data.rcF)rT verbose_evalr+)r(rc num_featuresnum_colnum_boosted_rounds feature_names feature_types)rrrresultrs r train_resultr{s Fii  g G    !T\\^ 33  % % ': 55  D$6$6 66  D$6$6 66 Mrc<eZdZdZdej dededefdZ y) ResetStrategyz"Callback for testing multi-output.repoch evals_logr cb|dzdk(r|jddiy|jddiy)Nrmulti_strategymulti_output_treeone_output_per_treeF) set_param)selfrrrs rafter_iterationzResetStrategy.after_iterations= 19> OO-/BC D OO-/DE FrN) __name__ __module__ __qualname____doc__r(Boosterrdictboolrrrrrs+,S[[RVrr)(rr functoolsrrtypingrrrnumpyr,xgboostr(xgboost.testingrjreXGBModelrrstrrDrrurrCrrriinfoint32maxrrrrrrcallbackTrainingCallbackrrrrrs -""%3s3t3nTDSTDDTDTTDn22JJ2.0jj2BE2 2&B1sB1dB1tB1J999RXXbhh  # # KH KHKH"%KH-0KH:=KHLOKH KH\     # -0 ?B   F S>!$:= #s(^*CLL11r