o i+@sdZddlZddlZddlZddlmZddlmZddl 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 die5d7fgd8d9Z6ej'(de%d:d;Z7dd?Z9dS)@z+Testing for Gaussian process classificationN) approx_fprime)ConvergenceWarning)GaussianProcessClassifier)RBFCompoundKernel WhiteKernel)ConstantKernel) MiniSeqKernel)assert_almost_equalassert_array_equalcCs t|S)N)npsin)xra/home/kim/smarthome/.venv/lib/python3.10/site-packages/sklearn/gaussian_process/tests/test_gpc.pyfs r )@g@g@g@g@Zdtypegffffffֿgffffff??fixed length_scalelength_scale_boundsg?)r)MbP?@@g{Gz?Y@cCsg|]}|tkr|qSr) fixed_kernel).0kernelrrr /sr$r#cCs<t|dtt}t|t|tdddfdkdS)Nr#r?)rfitXyr predict predict_probar#gpcrrrtest_predict_consistent2s*r.cCs\gd}tgd}tdd}t|d||}t||||dddfdkdS)NAZABBTFTrZbaseline_similarity_boundsr%rr&)r arrayr rr'r r*r+r(r)r#r-rrr"test_predict_consistent_structured9s  *r6cCs4t|dtt}||jj||jksJdS)Nr%)rr'r(r)log_marginal_likelihoodkernel_thetar,rrrtest_lml_improvingBsr:cCs0t|dtt}t||jj|ddS)Nr%)rr'r(r)r r7r8r9r,rrrtest_lml_precomputedKsr<cCsJt|dtt}tj|jjjtj d}|j |ddt |jj|ddS)Nr%rF)Z clone_kernelr;) rr'r(r)r Zonesr8r9shapeZfloat64r7r )r#r-Z input_thetarrrtest_lml_without_cloning_kernelTsr>cCszt|dtt}||jjd\}}tt |dk|jj|jj dddfkB|jj|jj dddfkBs;JdS)Nr%T-C6?rr) rr'r(r)r7r8r9r allabsbounds)r#r-lml lml_gradientrrrtest_converged_to_local_maximum^s  rEcsJt|dtt|jd\}}t|jfddd}t||ddS)Nr%Tcs |dS)NF)r7)r9r-rrss z#test_lml_gradient..g|=)rr'r(r)r7r9rr )r#rCrDZlml_gradient_approxrrFrtest_lml_gradientls rIc Csd\}}tj|}|||dd}t|jddtd|jdddk}tddtd g|d g|d }tj }t d D]#}t |||d  ||} | | j j} | |ttjjksdJ| }qCdS)N)rrr)ZaxisrHrrrr)r?r r)r#n_restarts_optimizerZ random_state)r random RandomStateZrandnr sumCrinfrangerr'r7r8r9ZfinfoZfloat32eps) global_random_seedZ n_samplesZ n_featuresrngr(r)r#Zlast_lmlrLgprCrrrtest_random_startsys( ,  rWcsFfdd}t||d}|tt||jj||jks!JdS)Nc stj}|||dd}}tdD].}t|td|dddftd|dddf}||dd}||krA||}}q||fS)NF)Z eval_gradientrrr)r rMrNrRZ atleast_1duniformmaximumminimum) Zobj_funcZ initial_thetarBrUZ theta_optZfunc_min_r9rrTrr optimizers   2  z(test_custom_optimizer..optimizer)r#r^)rr'r(y_mcr7r8r9)r#rTr^r-rr]rtest_custom_optimizers     r`cCsPt|d}|tt|t}t|dd|t}t t |d|dS)Nr%r) rr'r(r_r+X2r rOr*r r Zargmax)r#r-y_probZy_predrrrtest_multi_classs    rccCsPt|d}|ttt|dd}|tt|t}|t}t||dS)Nr%r)r#Zn_jobs)rr'r(r_r+rar )r#r-Zgpc_2rbZy_prob_2rrrtest_multi_class_n_jobss      rdc Cstddgd}t|d}d}tjt|d|ttWdn1s&wYtddgdtdd gd}t|d}t j d d G}t d |ttt |d ksXJt |djtsbJ|djjddksnJt |djtsxJ|djjddksJWdn1swYttd }tddgddgd}t|d}t j d d H}t d ||tt |d ksJt |djtsJ|djjddksJt |djtsJ|djjddksJWddS1swYdS)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_sumrgZX_tileZ kernel_dimsZgpc_dimsrrrtest_warning_boundss@        "rrzparams, error_type, err_msgz!kernel cannot be a CompoundKernelcCsNtdi|}tj||d|ttWddS1s wYdS)z0Check that expected error are raised during fit.reNr)rriraisesr'r(r))paramsZ error_typeerr_msgr-rrrtest_gpc_fit_errors "rvcCsTt|d}|tt|t\}}|jtjdfksJ|jtjdfks(JdS)>Checks that the latent mean and variance have the right shape.r%rN)rr'r(r)latent_mean_and_variancer=)r#r-Z latent_meanZlatent_variancerrr'test_gpc_latent_mean_and_variance_shape s  rycCsVttd}|tttjtdd|tWddS1s$wYdS)rwr%zdReturning the mean and variance of the latent function f is only supported for binary classificationreN) rrr'r(r_rirs ValueErrorrxrFrrrAtest_gpc_latent_mean_and_variance_complain_on_more_than_2_classes,s   "r{cCs@gd}tgd}tdd}t|d||}||dS)Nr/r2rr3r%)r r4r rr'rxr5rrr9test_latent_mean_and_variance_works_on_structured_kernels:s  r|):__doc__rjnumpyr riZscipy.optimizerZsklearn.exceptionsrZsklearn.gaussian_processrZ sklearn.gaussian_process.kernelsrrrrrPZ4sklearn.gaussian_process.tests._mini_sequence_kernelr Zsklearn.utils._testingr r rZ atleast_2dZlinspaceTr(rar4Zravelintr)ZfXemptyr=r_r!ZkernelsZnon_fixed_kernelsmarkZ parametrizer.r6r:r<r>rErIrWr`rcrdrrrzrvryr{r|rrrrsz                         I