o
    h                     @   s   d dl Z d dlZd dlZejjejjgZd dlmZ d dlm	Z	 d dl
mZ d dlmZ d dlmZ d dlmZ ejdd	d
d Zdd Zdd ZdS )    N)train_lstm_model)train_transformer_model)LemmaClassifier)evaluate_modelTEST_WORKING_DIR)convert_english_datasetmodule)scopec                   C   s
   t  dS )Nz/in/tiny_emb.ptr    r   r   f/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/tests/lemma_classifier/test_training.pypretrain_file   s   
r   c           	      C   s^   t | }t| d }|d }|d }d|d|d|d|g}t|}t|j| t|d }d S )Nlemma.ptr      z--wordvec_pretrain_file--save_name--train_file--eval_file)r   strr   mainr   modelr   load	tmp_pathr   converted_files	save_name
train_file	eval_file
train_argstrainerr   r   r   r   test_train_lstm   s   
r   c           	      C   s^   t | }t| d }|d }|d }ddd|d|d|g}t|}t|j| t|d }d S )	Nr   r   r   z--bert_modelzhf-internal-testing/tiny-bertr   r   r   )r   r   r   r   r   r   r   r   r   r   r   r   test_train_transformer%   s   
r    )globospytestmarkpipelinetravis
pytestmarkstanza.models.lemma_classifierr   r   )stanza.models.lemma_classifier.base_modelr   .stanza.models.lemma_classifier.evaluate_modelsr   stanza.testsr   3stanza.tests.lemma_classifier.test_data_preparationr   fixturer   r   r    r   r   r   r   <module>   s    

