o
    h                     @   sH   d dl Z d dlmZ d dlmZ dd Zd	ddZedkr"e  dS dS )
    N)Trainer)LemmaClassifierc                 C   s8   t | d}|D ]}t|}|j| q|| d S )N)
model_file)r   r   loadcontextual_lemmatizersappendsave)input_filenameoutput_filenameclassifierstrainer
classifier r   f/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/models/lemma/attach_lemma_classifier.pyattach_classifier   s
   

r   c                 C   s`   t  }|jdtddd |jdtddd |jdtddd	d
 || } t| j| j| j d S )Nz--inputTzWhich lemmatizer to start from)typerequiredhelpz--outputzWhere to save the lemmatizerz--classifier+zLemma classifier to attach)r   r   nargsr   )	argparseArgumentParseradd_argumentstr
parse_argsr   inputoutputr   )argsparserr   r   r   main   s   
r   __main__)N)r   stanza.models.lemma.trainerr   )stanza.models.lemma_classifier.base_modelr   r   r   __name__r   r   r   r   <module>   s    
		
