
    h"                     n    S SK Jr  S SKrS SKrS SKJr  S SKJr  S SKJ	r	   " S S5      r
 " S S	\5      rg)
    )AnyN)	NodeProto)Base)expectc                   4    \ rS rSrS\SS4S jrS\4S jrSrg)TfIdfVectorizerHelper   paramsreturnNc                     SnSnSnSnSnSnSnUUUUUUU/n	U	 H  n
X;   a  M
   SU
 35       e   X   U l         X   U l        X   U l        X   U l        X   U l        X   U l        X   U l        g )	Nmodemin_gram_lengthmax_gram_lengthmax_skip_countngram_countsngram_indexespool_int64szMissing attribute: r   r   r   r   r   r   r   )selfr
   r   r   r   r   r   r   r   required_attris              e/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/backend/test/case/node/tfidfvectorizer.py__init__TfIdfVectorizerHelper.__init__   s    ++)%'# 
 A;9"5aS 99;  L	%6%6$4"0#2!.    c                     [         R                  R                  SS/S/U R                  U R                  U R
                  U R                  U R                  U R                  U R                  S9
$ )NTfIdfVectorizerXY)	inputsoutputsr   r   r   r   r   r   r   )
onnxhelper	make_noder   r   r   r   r   r   r   )r   s    r   make_node_noweights)TfIdfVectorizerHelper.make_node_noweights/   sh    {{$$5E 00 00..**,,(( % 
 	
r   )r   r   r   r   r   r   r   )	__name__
__module____qualname____firstlineno__r   r   r   r%   __static_attributes__ r   r   r   r      s!    / / />
Y 
r   r   c                       \ rS rSr\SS j5       r\SS j5       r\SS j5       r\SS j5       r\SS j5       r	\SS j5       r
\SS	 j5       rS
rg)r   >   Nc            
      p   [         R                  " / SQ5      R                  [         R                  5      n [         R                  " / SQ5      R                  [         R                  5      n[         R                  " SS/5      R                  [         R
                  5      n[         R                  " / SQ5      R                  [         R
                  5      n[         R                  " / SQ5      R                  [         R
                  5      n[        SSSSUUUS	9nUR                  5       n[        UU /U/S
S9  g )N   r1      r2   r2            r3      r5   r4           r8   r8   r8         ?r9   r9   r      r   r1      r2   r:   r6   r5   
r<   r2   r6   r:   r6   r5   r3   r4   r5   r3   TFr<   r   *test_tfidfvectorizer_tf_only_bigrams_skip0r    r!   name	nparrayastypeint32float32int64r   r%   r   inputoutputr   r   r   r#   nodes          r   export_tf_only_bigrams_skip0,TfIdfVectorizer.export_tf_only_bigrams_skip0?   s    =>EEbhhO=>EEbjjQxxA'..rxx8!67>>rxxHhh=>EEHH
 '%'#
 ))+7H=		
r   c            
         [         R                  " / SQ/ SQ/5      R                  [         R                  5      n [         R                  " / SQ/ SQ/5      R                  [         R                  5      n[         R                  " SS/5      R                  [         R
                  5      n[         R                  " / SQ5      R                  [         R
                  5      n[         R                  " / SQ5      R                  [         R
                  5      n[        S	S
S
SUUUS9nUR                  5       n[        UU /U/SS9  g )Nr1   r1   r2   r2   r2   r3   r4   r5   r3   r6   r5   r4   r8   r8   r8   r8   r8   r8   r8   )r8   r8   r8   r8   r9   r8   r9   r   r:   r;   r=   r>   r<   r   /test_tfidfvectorizer_tf_batch_onlybigrams_skip0r@   rB   rI   s          r   !export_tf_batch_onlybigrams_skip01TfIdfVectorizer.export_tf_batch_onlybigrams_skip0[       ,.@ABII"((S02UV

&
 	 xxA'..rxx8!67>>rxxHhh=>EEHH
 '%'#
 ))+7HB		
r   c            
      p   [         R                  " / SQ5      R                  [         R                  5      n [         R                  " / SQ5      R                  [         R                  5      n[         R                  " SS/5      R                  [         R
                  5      n[         R                  " / SQ5      R                  [         R
                  5      n[         R                  " / SQ5      R                  [         R
                  5      n[        SSSSUUUS9nUR                  5       n[        UU /U/S	S
9  g )Nr0   )r9   r9   r9   r   )r   r1   r<   )r6   r5   r3   r4   r5   r3   r>   r<   r   .test_tfidfvectorizer_tf_onlybigrams_levelemptyr@   rB   rI   s          r    export_tf_onlybigrams_levelempty0TfIdfVectorizer.export_tf_onlybigrams_levelemptyy   s    =>EEbhhO/*11"**=xxA'..rxx8+22288<hh1299HH
 '%'#
 ))+7HA		
r   c            
      p   [         R                  " / SQ5      R                  [         R                  5      n [         R                  " / SQ5      R                  [         R                  5      n[         R                  " SS/5      R                  [         R
                  5      n[         R                  " / SQ5      R                  [         R
                  5      n[         R                  " / SQ5      R                  [         R
                  5      n[        SSSS	UUUS
9nUR                  5       n[        UU /U/SS9  g )Nr0   )r8   r8   r8   r8   r9         @r9   r   r:   r;   r=   r>   r<   r6   r   )test_tfidfvectorizer_tf_onlybigrams_skip5r@   rB   rI   s          r   export_tf_onlybigrams_skip5+TfIdfVectorizer.export_tf_onlybigrams_skip5   s    =>EEbhhO=>EEbjjQxxA'..rxx8!67>>rxxHhh=>EEHH
 '%'#
 ))+7H<		
r   c            
         [         R                  " / SQ/ SQ/5      R                  [         R                  5      n [         R                  " / SQ/ SQ/5      R                  [         R                  5      n[         R                  " SS/5      R                  [         R
                  5      n[         R                  " / SQ5      R                  [         R
                  5      n[         R                  " / SQ5      R                  [         R
                  5      n[        S	S
S
SUUUS9nUR                  5       n[        UU /U/SS9  g )NrP   rQ   rR   r7   r   r:   r;   r=   r>   r<   r6   r   /test_tfidfvectorizer_tf_batch_onlybigrams_skip5r@   rB   rI   s          r   !export_tf_batch_onlybigrams_skip51TfIdfVectorizer.export_tf_batch_onlybigrams_skip5   rV   r   c            
      p   [         R                  " / SQ5      R                  [         R                  5      n [         R                  " / SQ5      R                  [         R                  5      n[         R                  " SS/5      R                  [         R
                  5      n[         R                  " / SQ5      R                  [         R
                  5      n[         R                  " / SQ5      R                  [         R
                  5      n[        SSS	S
UUUS9nUR                  5       n[        UU /U/SS9  g )Nr0   )r8   r\   r9   r8   r9   r\   r9   r   r:   r;   r=   r>   r1   r<   r6   r   +test_tfidfvectorizer_tf_uniandbigrams_skip5r@   rB   rI   s          r   export_tf_uniandbigrams_skip5-TfIdfVectorizer.export_tf_uniandbigrams_skip5   s    =>EEbhhO=>EEbjjQxxA'..rxx8!67>>rxxHhh=>EEHH
 '%'#
 ))+7H>		
r   c            
         [         R                  " / SQ/ SQ/5      R                  [         R                  5      n [         R                  " / SQ/ SQ/5      R                  [         R                  5      n[         R                  " SS/5      R                  [         R
                  5      n[         R                  " / SQ5      R                  [         R
                  5      n[         R                  " / SQ5      R                  [         R
                  5      n[        S	S
SSUUUS9nUR                  5       n[        UU /U/SS9  g )NrP   rQ   )r8   r\   r8   r8   r8   r8   r8   )r8   r8   r9   r8   r9   r9   r9   r   r:   r;   r=   r>   r1   r<   r6   r   1test_tfidfvectorizer_tf_batch_uniandbigrams_skip5r@   rB   rI   s          r   #export_tf_batch_uniandbigrams_skip53TfIdfVectorizer.export_tf_batch_uniandbigrams_skip5   s    ,.@ABII"((S02UV

&
 	 xxA'..rxx8!67>>rxxHhh=>EEHH
 '%'#
 ))+7HD		
r   r,   )r   N)r'   r(   r)   r*   staticmethodrM   rT   rY   r^   rb   rf   rj   r+   r,   r   r   r   r   >   s    
 
6 
 
: 
 
6 
 
6 
 
: 
 
6 
 
r   r   )typingr   numpyrC   r"   r   onnx.backend.test.case.baser   onnx.backend.test.case.noder   r   r   r,   r   r   <module>rq      s2   
     , .,
 ,
^I
d I
r   