U
    ½mœdè  ã                   @   sÔ   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 d dlmZ d dlmZ edd d	�\ZZe e¡Ze e¡Ze eeg¡jZd
d„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Z dS )é    N)Úmake_regression)ÚRidge)ÚKernelRidge)Úpairwise_kernels)Úignore_warnings)Úassert_array_almost_equalé
   )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   úX/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/sklearn/tests/test_kernel_ridge.pyÚtest_kernel_ridge   s    r   c                  C   sD   t dddd� tt¡ t¡} tddd� tt¡ t¡}t| |ƒ d S ©Nr
   FZcholesky)r   r   Zsolverr   r   )r   r   ÚXcsrr   r   r   r   r   r   r   r   Útest_kernel_ridge_csr   s     ÿþÿr   c                  C   sD   t dddd� tt¡ t¡} tddd� tt¡ t¡}t| |ƒ d S r   )r   r   ÚXcscr   r   r   r   r   r   r   r   Útest_kernel_ridge_csc#   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 ]L} 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_precomputed7   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_unchanged?   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_weightsF   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_outputU   s    
r2   )!Únumpyr&   Zscipy.sparseÚsparseÚspZsklearn.datasetsr   Zsklearn.linear_modelr   Zsklearn.kernel_ridger   Zsklearn.metrics.pairwiser   Zsklearn.utils._testingr   r   r   r   Z
csr_matrixr   Z
csc_matrixr   r1   r(   r0   r   r   r   r!   r%   r*   r/   r2   r   r   r   r   Ú<module>   s&   




