
    Ng                     8    d dl Z d dlmZ d Zd Zd Zd Zd ZdS )    N)hmmc            	          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          |          z                      }t          j        | ||||          }|| ||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models          S/var/www/html/ai-engine/env/lib/python3.11/site-packages/nltk/test/unit/test_hmm.py_wikipedia_example_hmmr      s     i F=)G
sc3Z A
sc3Z A
sB
I
I
IC
s3S)**
+
+C"67Aq"==E&'3&&    c                      ddl m}  t          j                    \  }}}ddg}g dg dg}d|                    |          z  } | ||           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fps          r   test_forward_probabilityr!      s}    777777 !466E67
&C"""$<$<$<=H	
e((--	-Bb(+++++r   c                      ddl m}  t                      \  }}}}d|                    |          z  }|j        |                    d          z  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_resultss          r   test_forward_probability2r*   %   s    777777"8":":E67C	
e((--	-B $Q
	"B 
				 /Q77777r   c                      ddl m}  t                      \  }}}}d|                    |          z  }|j        |                    d          z  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)   s          r   test_backward_probabilityr.   9   s    777777"8":":E67C	
e))#..	.B $Q
	"B 
				c
	 /Q77777r   c                 .    t          j        d           d S )Nnumpy)pytestimportorskip)modules    r   setup_moduler4   Q   s    
     r   )r1   nltk.tagr   r   r!   r*   r.   r4    r   r   <module>r7      su          ' ' '$
, 
, 
,8 8 8(8 8 80! ! ! ! !r   