o
    
i&
                     @   s"  d dl Z d dlmZ d dlmZ d dlmZ dZe ed Z	ddd	d
difddd
d	difgZ
ddd	d	d
difddd
d	d	difgZe jdde
dde	dfdedde	dfde
dddddfdedddddfde
ddddddddd d!	fdedddddd"ddd#d!	fgd$d% ZdS )&    N)English)Example)Configz
[model]
@architectures = "spacy-legacy.HashEmbedCNN.v1"
pretrained_vectors = null
width = 96
depth = 4
embed_size = 2000
window_size = 1
maxout_pieces = 3
subword_features = true
modelzI'm so happy.catsg      ?g        )POSITIVENEGATIVEzI'm so angryzI'm angry and confused)ANGRYCONFUSEDHAPPYzI'm confused but happyzname,train_data,textcat_configtextcatzspacy-legacy.TextCatCNN.v1T)@architecturesexclusive_classestok2vectextcat_multilabelFzspacy-legacy.TextCatBOW.v1   )r   r   
ngram_sizeno_output_layer   zspacy-legacy.TextCatEnsemble.v1             g?)	r   width
embed_sizer   r   window_size
conv_depthpretrained_vectorsdropout   g?c                 C   s   d|i}t  }|j| |d}g }|D ]!\}}|t||| |d D ]	\}	}
||	 q)q|	 }t
dD ]}i }|j|||d q<d S )Nr   )configr      )sgdlosses)r   add_pipeappendr   	from_dictmake_docgetitems	add_label
initializerangeupdate)name
train_datatextcat_configpipe_confignlpr   train_examplestextannotationslabelvalue	optimizerir#    r:   \/home/ubuntu/.local/lib/python3.10/site-packages/spacy_legacy/tests/pipeline/test_textcat.pytest_textcat   s   r<   )pytestspacy.lang.enr   spacy.trainingr   	thinc.apir   default_tok2vec_configfrom_strDEFAULT_TOK2VEC_MODELTRAIN_DATA_SINGLE_LABELTRAIN_DATA_MULTI_LABELmarkparametrizer<   r:   r:   r:   r;   <module>   s.    