o
    i                     @   sN   d dl mZmZ ddlmZ ddlmZ de fddZd	d
 Z	dddZ
dS )    )Modelnoop   )registry   )ParserStepModelg?c                 C   sR   |du r
d}t  }nd}tdtd| di| ||g| ||dt|t||ddS )	z(Set up a stepwise transition-based modelNFTparser_modelnI)tok2veclowerupper)	has_upperunseen_classesresize_output)nameforwarddimslayersrefsinitattrs)r   r   r   maybe_get_dimr   set)r
   r   r   r   dropoutr   r    r   I/home/ubuntu/.local/lib/python3.10/site-packages/spacy/ml/tb_framework.pyTransitionModel   s    
r   c                 C   s*   t || j| jd || jd d}||jfS )Nr   r   )r   trainr   )r   r   r   finish_steps)modelXis_train
step_modelr   r   r   r       s   
r   Nc                 C   s\   |  dj|d |  d}|  | jd r,| jd|d}|  dj|d d S d S )Nr
   )r    r   r   r   nOr   )get_ref
initializer   opsalloc2fget_dim)r   r    Yr   	statevecsr   r   r   r   ,   s   

r   )NN)	thinc.apir   r   utilr   r   r   r   r   r   r   r   r   r   r   <module>   s    
