o
    h                     @   sV   d dl mZ d dlmZ 	 G dd deZG dd deZG dd deZd	d
 ZdS )    )Enum)nnc                   @   s   e Zd ZdZdZdZdZdS )WVType            N)__name__
__module____qualname__WORD2VECGOOGLEFASTTEXTOTHER r   r   Z/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/models/classifiers/utils.pyr      s
    r   c                   @   s   e Zd ZdZdZdZdS )ExtraVectorsr   r   r   N)r	   r
   r   NONECONCATSUMr   r   r   r   r      s    r   c                   @   s   e Zd ZdZdZdS )	ModelTyper   r   N)r	   r
   r   CNNCONSTITUENCYr   r   r   r   r      s    r   c                 C   sD   g }| }|D ]}| t|| |}q| t|| t|S )z
    Build a sequence of fully connected layers to go from the final conv layer to num_classes

    Returns an nn.ModuleList
    )appendr   Linear
ModuleList)fc_input_size	fc_shapesnum_classes	fc_layersprevious_layer_sizeshaper   r   r   build_output_layers   s   
r"   N)enumr   torchr   r   r   r   r"   r   r   r   r   <module>   s    	