o
    \i                  	   @   s   d dl Zd dlZd dlmZ d dlmZmZmZ dd Z	ej
deg dg dgeg d	g d
gfeg dgjeg dgjfgdd Zdd ZdS )    N)HalfPoissonLoss)binary_log_losslog_losspoisson_lossc                  C   s@   t g dgj} t g dgj}t| |}t |sJ d S )Nr   r      ?      ?r
   )nparrayTr   isfinitey_truey_probloss r   /var/www/www-root/data/www/176.119.141.140/sports-predictor/venv/lib/python3.10/site-packages/sklearn/neural_network/tests/test_base.py"test_binary_log_loss_1_prob_finite   s   
r   zy_true, y_prob)r   r   r   )r   r   r   )        r
   r   )r	   皙?r   r   r   c                 C   s   t | |}t|sJ d S )N)r   r   r   r   r   r   r   test_log_loss_1_prob_finite   s   
r   c           
      C   s   d}t j| }|jdd|dt}||}t |}|jdd|d}d|v s+J t	|||d}t
 }||||d|j||d |   }	|tj|	d	d
ksTJ dS )z6Test Poisson loss against well tested HalfPoissonLoss.i  r   
   )lowhighsizeg?)r   y_predsample_weight)r   raw_predictionr   )r   r   g-q=)relN)r   randomdefault_rngintegersastypefloatstandard_normalexpuniformr   r   constant_to_optimal_zeromeanpytestapprox)
global_random_seednrngr   y_rawr   swr   plloss_refr   r   r   test_poisson_loss!   s    

r4   )numpyr   r+   sklearn._lossr   sklearn.neural_network._baser   r   r   r   markparametrizer   r   r   r4   r   r   r   r   <module>   s    	"

