o
    Mf                     @   s@   d dl Z d dlmZ dd Zdd Zdd Zd	d
 Zdd ZdS )    N)hmmc                  C   sv   ddg} ddg}ddgddgg}ddgd	d
gg}ddg}g d}t t|d gt| }t| ||||}|| ||fS )Nrainzno rainumbrellano umbrellagffffff?g333333?g?g?g?g?      ?)r   r   r   r   r   )listziplenr   _create_hmm_tagger)statessymbolsABpiseqmodel r   Y/var/www/html/analyze/labelStudio/lib/python3.10/site-packages/nltk/test/unit/test_hmm.py_wikipedia_example_hmm   s   r   c                  C   sN   ddl m}  t \}}}ddg}g dg dg}d|| }| || d S )Nr   assert_array_almost_equal)upN)gffffff?g{Gz?g
ףp=
?)gǺ?g rh?g48EG?   )numpy.testingr   r   _market_hmm_example_forward_probability)r   r   r   r   r   expectedfpr   r   r   test_forward_probability   s   r   c                  C   sn   ddl m}  t \}}}}d|| }|j|jdd j}ddgdd	gd
dgddgddgg}| ||d d S )Nr   r   r      axisgX2ı.?g48E?g0*D?g}bٽ?gqh?g?gmV}b?g&S:?gC?gHP?   )r   r   r   r   Tsum)r   r   r   r   r   r   wikipedia_resultsr   r   r   test_forward_probability2%   s   r&   c                  C   sn   ddl m}  t \}}}}d|| }|j|jdd j}ddgdd	gd
dgddgddgg}| ||d d S )Nr   r   r   r   r    g2w-!?g?g_L?g|гY?gAf?g|a2U0?gPs?g_Q?r   r"   )r   r   r   _backward_probabilityr#   r$   )r   r   r   r   r   bpr%   r   r   r   test_backward_probability9   s   r)   c                 C   s   t d d S )Nnumpy)pytestimportorskip)moduler   r   r   setup_moduleQ   s   r.   )r+   nltk.tagr   r   r   r&   r)   r.   r   r   r   r   <module>   s    