
    h                     @    S SK Jr  S SKr " S S5      r " S S5      rg)    )AnyNc                   >    \ rS rSrS rS\S\SS4S jrS\4S jrS	r	g)
SVMAttributes   c                     / U l         g )N_names)selfs    c/var/www/fran/franai/venv/lib/python3.13/site-packages/onnx/reference/ops/aionnxml/op_svm_helper.py__init__SVMAttributes.__init__   s	        namevaluereturnNc                     [        U[        5      (       aS  US;  aM  US;   a$  [        R                  " U[        R                  S9nO#[        R                  " U[        R
                  S9n[        XU5        g )N>   kernel_params>   vectors_per_classdtype)
isinstancelistnparrayint64float32setattr)r
   r   r   s      r   addSVMAttributes.add   sR    eT""t3D'D,,bhh7bjj9E"r   c           
          S/nU R                    H#  nUR                  SU S[        X5       35        M%     SR                  U5      $ )N
Attributesz  =
)r	   appendgetattrjoin)r
   rowsr   s      r   __str__SVMAttributes.__str__   sD    ~KKDKK"TF!GD$7#89:  yyr   r   )
__name__
__module____qualname____firstlineno__r   strr   r   r(   __static_attributes__ r   r   r   r      s-    # #C #D # r   r   c                       \ rS rSrSrS rS\4S jrS\R                  S\R                  S\S\R                  4S	 jr
S
\R                  S\R                  4S jrSrg)	SVMCommon   z
Base class for SVM.
c                    [        5       U l        UR                  5        H   u  p#U R                  R                  X#5        M"     U R                  R                  (       ad  U R                  R                  S   U l        U R                  R                  S   U l        [        U R                  R                  S   5      U l        g SU l        SU l        SU l        g )Nr                 )	r   attsitemsr   r   gamma_coef0_intdegree_)r
   kwargsr   r   s       r   r   SVMCommon.__init__#   s    !O	!<<>KDIIMM$& * 99""))11!4DK))11!4DKtyy66q9:DLDKDKDLr   r   c                 n    SSU R                    3[        U R                  5      /nSR                  U5      $ )NTreeEnsemblezroot_index=r#   )
root_indexr.   r8   r&   )r
   r'   s     r   r(   SVMCommon.__str__2   s0    +doo-> ?TYYPyyr   pApBkernelc                    UR                  5       nUS:X  a?  [        R                  " X5      nXPR                  -  U R                  -   nXPR
                  -  $ US:X  aG  [        R                  " X5      nXPR                  -  U R                  -   n[        R                  " U5      $ US:X  a:  X-
  nXf-  R                  5       n[        R                  " U R                  * U-  5      $ US:X  a  [        R                  " X5      $ [        SU< S35      e)NpolysigmoidrbflinearzUnexpected kernel=.)
lowerr   dotr:   r;   r=   tanhsumexp
ValueError)r
   rD   rE   rF   ksdiffs          r   
kernel_dotSVMCommon.kernel_dot6   s    LLN;rAKK$++-All?"	>rAKK$++-A771::7D!!#A664;;,*++=66">!-fZq9::r   Xc                 |   U R                   R                  S:  aT  SnU R                   R                  nU R                   R                  R	                  U R                   R                  S45      nOSnSn[
        R                  " UR                  S   S4UR                  S9n[        UR                  S   5       H  nSnUS:X  ax  [        U R                   R                  5       H9  nU R                  X   WU   U5      n	XpR                   R                  U   U	-  -  nM;     XpR                   R                  S   -  nOJUS:X  aD  U R                  X   U R                   R                  U5      nXpR                   R                  S   -  nU R                   R                  (       a  US:  a  SOSXVS4'   M  XuUS4'   GM     U$ )	Nr   SVM_SVC
SVM_LINEARLINEARr5   r   r7   )r8   
n_supportskernel_typesupport_vectorsreshaper   emptyshaper   rangerV   coefficientsrho	one_class)
r
   rX   mode_kernel_type_svznrT   jds
             r   run_regSVMCommon.run_regH   sp   99!#E9900L**22DII4H4H"3MNB E#LHHaggaj!_AGG4qwwqz"AA	!tyy334Abe\BA//2Q66A 5 YY]]1%%,&OOAD$))*@*@,OYY]]1%%yy"" 1u!"Q$!Q$ #  r   )r8   r;   r=   r:   N)r*   r+   r,   r-   __doc__r   r.   r(   r   ndarrayrV   ro   r/   r0   r   r   r2   r2      s`     ;RZZ ;RZZ ; ; ;$ 

 r   r2   )typingr   numpyr   r   r2   r0   r   r   <module>ru      s#      &E Er   