U
    _»{jÄ  ã                   @   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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)ÚKernelRidge)ÚRidge)Úpairwise_kernels)Úassert_array_almost_equalÚignore_warningsé
   )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   ú^/var/www/html/bwave/python/venv/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_precomputed4   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_weightsC   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_outputR   s    
r2   )!Únumpyr&   Zscipy.sparseÚsparseÚspZsklearn.datasetsr   Zsklearn.kernel_ridger   Zsklearn.linear_modelr   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$   




