o i8@s<dZddlmZddlZddlZddlmZddlm Z m Z m Z m Z m Z mZmZmZmZmZmZmZddlmZmZmZddlmZmZmZmZejd ddd Z!ejd ddd Z"d e!j#_$d e"j#_$e d d eddZ%e d d e dde ddd e dddd e dd e%d e dd gd d edddd edddd edddd edddd edd gddded dgdddeddgddedddeddde d de d dde d gd ed gd gZ&eD]Z'e'd vrqe&(ed!e'd"qej)*d#e&d$d%Z+ej)*d#d&d'e&Dd(d)Z,ej)*d#d*d'e&Dd+d,Z-ej)*d#e&d-d.Z.d/d0Z/d1d2Z0ej)*d#d3d'e&Dd4d5Z1ej)*d#e&d6d7Z2d8d9Z3d:d;Z4ej)*d#e&dd?Z6d@dAZ7ej)*d#e&dBdCZ8ej)*d#e&dDdEZ9ej)*d#e&dFdGZ:dHdIZ;dS)Jz+Testing for kernels for Gaussian processes.) signatureN)clone) RBFCompoundKernelConstantKernel DotProductExponentiationExpSineSquaredKernelOperatorMaternPairwiseKernelRationalQuadratic WhiteKernel_approx_fprime)PAIRWISE_KERNEL_FUNCTIONSeuclidean_distancespairwise_kernels)assert_allcloseassert_almost_equalassert_array_almost_equalassert_array_equal))rF@ length_scale@Z noise_level)?r)length_scale_boundsg$@)Zconstant_valuegQ?fixed)rr!r )rnu?@@)ralpha)rZ periodicity)Zsigma_0r)Z additive_chi2Zchi2?)gammametrickernelcsttdd\}}|jdtjdksJ|jdtjdks$J|jdjjdks1Jfdd}tj|d}t||d dS) NTZ eval_gradientrrrcs|}|tdd}|S)NFr,)Zclone_with_thetaX)thetaZ kernel_cloneKr+e/home/kim/smarthome/.venv/lib/python3.10/site-packages/sklearn/gaussian_process/tests/test_kernels.pyeval_kernel_for_thetaTs  z3test_kernel_gradient..eval_kernel_for_theta绽|=)rr-shaper.rr)r+r/ K_gradientr3ZK_gradient_approxr1r0r2test_kernel_gradientJs r8cCsg|] }t|ttfs|qSr1) isinstancer r.0r+r1r1r2 `s  r<c Cs4t|}|j}|tdd\}}t|jjj}dd|D}tddt dd|}t dd |j Dt |ks;Jt |j D]\}}||t t||jksSJq@t |j D]\}}|} d | |jd <|j} | di| } | tdd\}} |jd | jjd d ksJ|jd| jdd ksJ|d kr|d|| jd|ksJt|dd|f| dd|f|d t|j kr||d d| j|dksJt|d|d df| d|dfqYt |j D])\}}t d||<||_tt||jdt||jdt|j|t dqdS)NTr,cSsg|] }|jdkr|jqS)selfname)r;pr1r1r2r<osz%test_kernel_theta..cSs|dtd S)Nr_bounds)lensr1r1r2qsz#test_kernel_theta..cSs |dS)NrA)endswithrCr1r1r2rEqs css|]}|jVqdSNr>)r;hyperparameterr1r1r2 ssz$test_kernel_theta..r"rArrr.*+r1)rr.r-r __class____init__ parametersvaluesmapfiltersethyperparameters enumeratenploggetattrr? get_paramsr6rrBrsetattr) r+r._r7Z init_signargsZ theta_varsirHparamsZ kernel_classZ new_kernelZK_gradient_newr1r1r2test_kernel_theta^sF   ""&r^cCsg|]}|tkr|qSr1)kernel_rbf_plus_whiter:r1r1r2r<s cCs*t|}|t}|tt}t||ddSNr)rr-r)r+ZK_autoZK_crossr1r1r2test_auto_vs_crosss  racCs0t|}t|t}|t}t||ddSr`)rrUdiagr-r)r+Z K_call_diagZK_diagr1r1r2test_kernel_diags rccCsHttddtdtdttdtdttddtdS)Nrr(r)rrr-r1r1r1r2 test_kernel_operator_commutatives"&rdcCsdtddg}|t}t}|dddfd9<dtd|}t||t}|dddfd<dtd|}t|||jtd|_t|jtgdt|jj d d gdS) Nrr rrr5rr)g@r(r&r(r&) rr-copyrr.rUrVrk2r)r+r/ZX1K1ZX2K2r1r1r2test_kernel_anisotropics  ricCsg|]}|r|qSr1)Z is_stationaryr:r1r1r2r<scCs.t|}|ttd}t|dt|dS)Nr)rr)rr-rrUrb)r+r/r1r1r2test_kernel_stationarysrjcCsPt|}t|tr|j|jjksJt|tr$|j|jjp |jjks&JdSdSrG)rr9rrequires_vector_inputr+r Zk1rfr0r1r1r2test_kernel_input_types  rlcCs@ttddg}|jr Jttddtddg}|jsJdS)Nrrrr)rrrkrr0r1r1r2test_compound_kernel_input_types rmcCsHtt|t|D]}|dr!t||}t||}||ks!Jq dS)NZhyperparameter_)rRdir startswithrW)Zkernel1Zkernel2attrZ attr_value1Z attr_value2r1r1r2check_hyperparameters_equals    rqcCsRt|}t|}||ksJt|t|ksJ||ks"Jt||dSrG)ridrXrq)r+ kernel_clonedr1r1r2test_kernel_clones  rtcCst|}d}t|}|}ttf}d|vrct||se|d}t|r/|d|d<||d<n |gd|d<|d|d<|jdi|t|}||ksRJt|t|ks\Jt ||dSdSdS)N)gh㈵>gj@rrr!rr1) rrXr r r9rUiterable set_paramsrrrq)r+boundsrsr]Zisotropic_kernelsrZkernel_cloned_cloner1r1r2"test_kernel_clone_after_set_params s$    rxcCstdddt}tt|ttjdttttdd }tdddt}t||t ddt}ttj ddt}t||t ||d }d D]}t|ddt}t||ddt}t||qLd }t|ddt}t ddt}t||d d dS)Nr$r()r#rrF)Zsquaredr rr4)r r$r%dr)decimal) r r-rrUrbZonesr6exprrinfr)r/ZK_absexpZK_rbfZtinyr#rgrhZlarger1r1r2test_matern_kernel(s$    r}cCsTt|}|tkr|t}tt|d}t|||tt}ttt|d}t||dS)N)r*)rr_r-rrY)r+rgrhr1r1r2test_kernel_versus_pairwiseEs   rcCs\t|}d}|}|jD]@}tdt|jr|jdkrq |j}|dkr;tt |j |||||j ||7}q tt |j |||j |d7}q d}d}|jD]V}tdt|jre|jdkreqU|j}|dkr|j di|j |g|itt |j ||||g|||7}qU|j di|j |itt |j |||d7}qUdS)Nrstringr"r r1) rrXrSr9typerwZ n_elementsrrUr{r.r?rv)r+indexr]rHsizevaluer1r1r2test_set_get_paramsXsD        rcCst|}t|dSrG)rreprr0r1r1r2test_repr_kernelss rcCsNtddgd}d}tjt|d |tWddS1s wYdS)Nr(rzeRationalQuadratic kernel only supports isotropic version, please use a single scalar for length_scale)match)r pytestZraisesAttributeErrorr-)r+messager1r1r2test_rational_quadratic_kernels  "r)<__doc__inspectrnumpyrUrZ sklearn.baserZ sklearn.gaussian_process.kernelsrrrrrr r r r r rrZsklearn.metrics.pairwiserrrZsklearn.utils._testingrrrrrandomZ RandomStatenormalr-r~flagsZ writeabler_Zkernelsr*appendmarkZ parametrizer8r^rarcrdrirjrlrmrqrtrxr}rrrrr1r1r1r2s  8         2               *