o
    iH                     @   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
   )Z
n_featuresZ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   Y/home/kim/smarthome/.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   FZcholesky)r   r   Zsolverr   r   )r   r   r   r   r   r   r   )r   Z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   ZrbfZpolyZcosine)Z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"   ZK2r   r   r   .test_kernel_ridge_precomputed_kernel_unchanged3   s   r(   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   )Zsample_weightr   r   r!   )r$   r%   r   r&   randomZRandomStateZ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   r,   r   r   r   test_kernel_ridge_multi_outputI   s   
r0   )numpyr$   ZpytestZsklearn.datasetsr   Zsklearn.kernel_ridger   Zsklearn.linear_modelr   Zsklearn.metrics.pairwiser   Zsklearn.utils._testingr   r   Zsklearn.utils.fixesr   r	   r   r   r/   r&   r.   r   markZparametrizer   r   r#   r(   r-   r0   r   r   r   r   <module>   s$    

