o
    ?HhH                     @   s   d dl Zd dlZd dlmZ d dlmZ d dlmZ d dl	m
Z
 d dlmZmZ d dlmZmZ edd d	\ZZeeegjZd
d Zejdg eedd Zdd Zdd Zdd Zdd Zdd ZdS )    N)make_regression)KernelRidge)Ridge)pairwise_kernels)assert_array_almost_equalignore_warnings)CSC_CONTAINERSCSR_CONTAINERS
   )
n_featuresrandom_statec                  C   sB   t dddttt} tdddttt}t| | d S N   Falphafit_interceptlinearkernelr   )r   fitXypredictr   r   )predpred2 r   _/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/sklearn/tests/test_kernel_ridge.pytest_kernel_ridge   s   r   sparse_containerc                 C   sL   | t }tdddd|t|}tddd|t|}t|| d S )Nr   Fcholesky)r   r   solverr   r   )r   r   r   r   r   r   r   )r   X_sparser   r   r   r   r   test_kernel_ridge_sparse   s   r"   c                  C   sN   t dddttt} tddd}t|jtt |t}t| | d S )Nr   Fr   r   r   )r   r   r   r   r   r   r   r   )r   krr   r   r   r   !test_kernel_ridge_singular_kernel!   s
   
r$   c                  C   sV   dD ]&} t tt| d}t| dttt}tdd|t|}t|| qd S )N)r   rbfpolycosine)metricr   precomputed)r   r   r   r   r   r   r   )r   Kr   r   r   r   r   test_kernel_ridge_precomputed+   s   r,   c                  C   s6   t ttj} |  }tdd| t t| | d S )Nr*   r)   )	npdotr   Tcopyr   r   r   r   )r+   K2r   r   r   .test_kernel_ridge_precomputed_kernel_unchanged3   s   r2   c                  C   s   t ttj} t jdtjd }tdddj	tt
|dt}tdddj	tt
|dt}tdddj	| t
|d| }t|| t|| d S )	Nr   r   Fr   )sample_weightr   r   r*   )r-   r.   r   r/   randomRandomStaterandshaper   r   r   r   r   r   )r+   swr   r   pred3r   r   r    test_kernel_ridge_sample_weights:   s   

r:   c                  C   sv   t dddttt} tdddttt}t| | tdddttt}t	||gj
}t|| d S r   )r   r   r   Yr   r   r   r   r-   arrayr/   )r   r   r9   r   r   r   test_kernel_ridge_multi_outputI   s   
r=   )numpyr-   pytestsklearn.datasetsr   sklearn.kernel_ridger   sklearn.linear_modelr   sklearn.metrics.pairwiser   sklearn.utils._testingr   r   sklearn.utils.fixesr   r	   r   r   r<   r/   r;   r   markparametrizer"   r$   r,   r2   r:   r=   r   r   r   r   <module>   s$    

