3
Ðédñ  ã               @   s(  d dl Z d dlmZ d dlmZmZmZ d dlmZ d dlmZ d dlm	Z	 e j
jdded	d	gd
dgddœeddddœfded	dgddgddœedddœfded	ddddœeddddœfded	ddddœedddœfde	dddddddœeddddœfde	dddddddœedddœfgƒdd„ ƒZdS )é    N)Úutil)Ú
Tok2Vec_v1ÚMultiHashEmbed_v1ÚCharacterEmbed_v1)ÚMaxoutWindowEncoder_v1)ÚMishWindowEncoder_v1)ÚHashEmbedCNN_v1z7width,embed_arch,embed_config,encode_arch,encode_configé   éd   ZSHAPEZLOWERF)ÚrowsÚattrsÚinclude_static_vectorsé   é   é   )Úwindow_sizeÚmaxout_piecesÚdepthé   ZORTHÚPREFIXé   )r   r   é@   )r   ZnMZnCr   é   é   T)r   Z
embed_sizer   r   Zsubword_featuresZpretrained_vectorsc       	      C   s†   | |d< | |d< t jdƒ}t|f |Ž|f |Žƒ}|j|ƒ |j|ƒ\}}t|ƒt|ƒks\t‚|d jt|d ƒ| fkszt‚||ƒ d S )NÚwidthr   r   )r   Z	get_batchr   Z
initializeZbegin_updateÚlenÚAssertionErrorÚshape)	r   Z
embed_archZembed_configZencode_archZencode_configZdocsZtok2vecZvectorsZbackprop© r   úZ/var/www/html/virt/lib/python3.6/site-packages/spacy_legacy/tests/pipeline/test_tok2vec.pyÚtest_tok2vec	   s    

r    )ZpytestZspacy.testsr   Z"spacy_legacy.architectures.tok2vecr   r   r   r   r   r   ÚmarkZparametrizer    r   r   r   r   Ú<module>   s   $""