a h&@sZdZddlZddlZddlZddlmZddlmZddl m Z ddl m Z m Z mZddl mZddlmZdd lmZmZd d Zeedd d jZegdjZejeedkedZeeZ ej!ej"edZ#de#e dk<de#e dke dk@<de#e dk<e dddZ$e dde$e dddedde dddgZ%dde%DZ&ej'(de%ddZ)d d!Z*ej'(de&d"d#Z+ej'(de%d$d%Z,ej'(de%d&d'Z-ej'(de&d(d)Z.ej'(de%d*d+Z/d,d-Z0ej'(de&d.d/Z1ej'(de%d0d1Z2ej'(de%d2d3Z3d4d5Z4ej'(d6de die5d7fgd8d9Z6dS):z+Testing for Gaussian process classificationN) approx_fprime)ConvergenceWarning)GaussianProcessClassifier)RBFCompoundKernel WhiteKernel)ConstantKernel) MiniSeqKernel)assert_almost_equalassert_array_equalcCs t|S)N)npsin)xrc/var/www/html/assistant/venv/lib/python3.9/site-packages/sklearn/gaussian_process/tests/test_gpc.pyfsr )@g@g@g@g@Zdtypegffffffֿgffffff??fixed length_scalelength_scale_boundsg?)r)MbP?@@g{Gz?Y@cCsg|]}|tkr|qSr) fixed_kernel).0kernelrrr /r$r#cCs<t|dtt}t|t|tdddfdkdS)Nr#r?)rfitXyr predict predict_probar#gpcrrrtest_predict_consistent2sr/cCs\gd}tgd}tdd}t|d||}t||||dddfdkdS)N)AZABB)TFTr)Zbaseline_similarity_boundsr&rr')r arrayr rr(r r+r,)r)r*r#r.rrr"test_predict_consistent_structured9s  r3cCs4t|dtt}||jj||jks0JdS)Nr&)rr(r)r*log_marginal_likelihoodkernel_thetar-rrrtest_lml_improvingBsr7cCs0t|dtt}t||jj|ddS)Nr&)rr(r)r*r r4r5r6r-rrrtest_lml_precomputedKsr9cCsJt|dtt}tj|jjjtj d}|j |ddt |jj|ddS)Nr&rF)Z clone_kernelr8) rr(r)r*r Zonesr5r6shapeZfloat64r4r )r#r.Z input_thetarrrtest_lml_without_cloning_kernelTsr;cCszt|dtt}||jjd\}}tt |dk|jj|jj dddfkB|jj|jj dddfkBsvJdS)Nr&T-C6?rr) rr(r)r*r4r5r6r allabsbounds)r#r.lml lml_gradientrrrtest_converged_to_local_maximum^s rBcsJt|dtt|jd\}}t|jfddd}t||ddS)Nr&Tcs |dS)NF)r4)r6r.rrsr%z#test_lml_gradient..g|=)rr(r)r*r4r6rr )r#r@rAZlml_gradient_approxrrCrtest_lml_gradientls rFc Csd\}}tj|}|||dd}t|jddtd|jdddk}tddtd g|d g|d }tj }t d D]F}t |||d  ||} | | j j} | |ttjjksJ| }qdS)N)rrr)ZaxisrErrrr)r<r r)r#n_restarts_optimizerZ random_state)r random RandomStateZrandnr sumCrinfrangerr(r4r5r6ZfinfoZfloat32eps) global_random_seedZ n_samplesZ n_featuresrngr)r*r#Zlast_lmlrIgpr@rrrtest_random_startsys& ,  rTcsFfdd}t||d}|tt||jj||jksBJdS)Nc stj}|||dd}}tdD]\}t|td|dddftd|dddf}||dd}||kr&||}}q&||fS)NF)Z eval_gradientrrr)r rJrKrOZ atleast_1duniformmaximumminimum) Zobj_funcZ initial_thetar?rRZ theta_optZfunc_min_r6rrQrr optimizers   2  z(test_custom_optimizer..optimizer)r#r[)rr(r)y_mcr4r5r6)r#rQr[r.rrZrtest_custom_optimizers    r]cCsPt|d}|tt|t}t|dd|t}t t |d|dS)Nr&r) rr(r)r\r,X2r rLr+r r Zargmax)r#r.y_probZy_predrrrtest_multi_classs     r`cCsPt|d}|ttt|dd}|tt|t}|t}t||dS)Nr&r)r#Zn_jobs)rr(r)r\r,r^r )r#r.Zgpc_2r_Zy_prob_2rrrtest_multi_class_n_jobss      rac Cstddgd}t|d}d}tjt|d|ttWdn1sL0Ytddgdtdd gd}t|d}t j d d }t d |ttt |d ksJt |djtsJ|djjddksJt |djtsJ|djjddks JWdn1s 0Yttd }tddgddgd}t|d}t j d d }t d ||tt |d ksJt |djtsJ|djjddksJt |djtsJ|djjddksJWdn1s0YdS)Ngh㈵>r)rr&zThe optimal value found for dimension 0 of parameter length_scale is close to the specified upper bound 0.001. Increasing the bound and calling fit again may find a better value.match)Znoise_level_boundsrgj@T)recordalwaysrrzThe optimal value found for dimension 0 of parameter k1__noise_level is close to the specified upper bound 0.001. Increasing the bound and calling fit again may find a better value.rzThe optimal value found for dimension 0 of parameter k2__length_scale is close to the specified lower bound 1000.0. Decreasing the bound and calling fit again may find a better value.rrg$@r rzThe optimal value found for dimension 0 of parameter length_scale is close to the specified upper bound 100.0. Increasing the bound and calling fit again may find a better value.zThe optimal value found for dimension 1 of parameter length_scale is close to the specified upper bound 100.0. Increasing the bound and calling fit again may find a better value.)rrpytestZwarnsrr(r)r*rwarningscatch_warnings simplefilterlen issubclasscategorymessageargsr Ztile) r#r.Zwarning_messageZ kernel_sumZgpc_sumrdZX_tileZ kernel_dimsZgpc_dimsrrrtest_warning_boundssB *    (     rozparams, error_type, err_msgz!kernel cannot be a CompoundKernelcCsLtfi|}tj||d|ttWdn1s>0YdS)z0Check that expected error are raised during fit.rbN)rrfZraisesr(r)r*)paramsZ error_typeerr_msgr.rrrtest_gpc_fit_errors rr)7__doc__rgnumpyr rfZscipy.optimizerZsklearn.exceptionsrZsklearn.gaussian_processrZ sklearn.gaussian_process.kernelsrrrrrMZ4sklearn.gaussian_process.tests._mini_sequence_kernelr Zsklearn.utils._testingr r rZ atleast_2dZlinspaceTr)r^r2Zravelintr*ZfXemptyr:r\r!ZkernelsZnon_fixed_kernelsmarkZ parametrizer/r3r7r9r;rBrFrTr]r`raro ValueErrorrrrrrrsp                         I