
    -i                        S SK Jr  S SKrS SKrS SKJr  S SKJrJ	r	  S SK
Jr  S SKJr  S SKJrJrJr  SS	/rS
 rS r\R*                  R-                  S/ SQ5      \R*                  R-                  S\5      S 5       5       rS rS rS rS rg)    )productN)linalg)manifold	neighbors)
make_blobs)barycenter_kneighbors_graph)assert_allcloseassert_array_equalignore_warningsdensearpackc                 ,   [         R                  " SS/SS/SS//U S9n[        US5      n[         R                  " / SQ/ SQ/ SQ/U S9nUR                  U :X  d   e[	        UR                  5       U5        [        US5      n[	        [         R                  " UR                  5       SS	9[         R                  " S
5      5        [         R                  " UR                  5       U5      n[        R                  " XA-
  5      UR                  S   -  S:  d   eg )Nr      g)\(?      ?   )dtype)        r   r   )r   r   r   )axis   )nparrayr   r   r	   toarraysumonesdotr   normshape)global_dtypeXgraphexpected_graphpreds        ]/var/www/html/venv/lib/python3.13/site-packages/sklearn/manifold/tests/test_locally_linear.py test_barycenter_kneighbors_graphr$      s    
1a&4+1v.lCA'1-EXX	/?;<N ;;,&&&EMMO^4'1-EBFF5==?3RWWQZ@66%--/1%D;;tx 1771:-111    c           	         [         R                  R                  S5      n[         R                  " [	        [        [        S5      SS95      5      nUSUR                  UR                  S9-  -   nUR                  U SS9nSn[        R                  " SX1S	9nS
n[        X$R                  5      R                  5       n[        R                   " [         R"                  " Xb5      U-
  S5      nXu:  d   e[$         H  nUR'                  US9  UR)                  U5        UR*                  R                  S   U:X  d   e[        R                   " [         R"                  " XdR*                  5      UR*                  -
  S5      S-  nXu:  d   e[-        UR.                  US
S9  M     UR0                  " UR                  6 R                  U SS9S-  n	UR3                  X)-   5      n
[        R                   " XR*                  -
  5      U:  d   eg )N*      r   repeat绽|=sizeFcopy)n_neighborsn_componentsrandom_stateg?froeigen_solverr   )atold   )r   randomRandomStater   listr   rangeuniformr   astyper   LocallyLinearEmbeddingr   r0   r   r   r   r   eigen_solvers
set_paramsfit
embedding_r	   reconstruction_error_randn	transform)r   rngr   r1   clftolNreconstruction_errorsolvernoiseX_reembeddeds              r#   test_lle_simple_gridrN   ,   s    ))


#C 	geAhq123A	ECKKQWWK---A	E*AL

)
)LC C#A7??AA!;;rvva|a'7?%%%F+
~~##A&,666KKq..1CNNBEJaO 	 $)))113GdS   IIqww&&|%&@3FE==+L;;|nn45;;;r%   methodstandardhessianmodifiedltsarK   c           	         [         R                  R                  S5      n[         R                  " [	        [        [         R                  " S5      SS95      5      n[         R                  XDS S 2S4   S-  S-  4   nUSUR                  UR                  S9-  -   nUR                  U SS9nSn[        R                  " S	XQSS
9nUS:X  a  SOSn[        XFR                  5      R                  5       n[         R"                  " [         R$                  " X5      U-
  5      n	X:  d   eUR'                  US9  UR)                  U5        UR*                  R                  S   U:X  d   e[         R"                  " [         R$                  " XR*                  5      UR*                  -
  S5      S-  n	SU< SU< 3n
X:  d   U
5       e[         R,                  " UR.                  U	-
  5      Xy-  :  d   U
5       eg )Nr      r   r)   r+   r,   Fr.      )r0   r1   rO   r2   rQ   g      ?r   r4   r   r3   zsolver: z
, method: )r   r8   r9   r   r:   r   arangec_r<   r   r=   r   r>   r   r0   r   r   r   r   r@   rA   rB   absrC   )r   rO   rK   rF   r   r1   rG   rH   rI   rJ   detailss              r#   test_lle_manifoldr\   T   s    ))


"C
gbiimA678A
a1a4A""#A	ECKKQWWK---A	E*AL

)
)LaC :%#1C#A7??AA!;;rvva|a'78%%%NNN'GGAJ>>"l222BFF1nn->F!K  +1&9G%.w.%
s((+??@

$	% 	%r%   c                      SSK Jn Jn  U R                  SS9u  p#UR	                  S[
        R                  " SS94S[        R                  " 5       4/5      nUR                  X#5        SUR                  X#5      :  d   eg )Nr   )datasetspipeline)r2   filterrG   g?)sklearnr^   r_   r   Pipeliner   r>   r   KNeighborsClassifierrA   score)r^   r_   r   yrG   s        r#   test_pipelinerf   v   sx     +A.DA


x66AFGI2245	
C GGAM1   r%   c                      [         R                  " S5      n [        n[        R                  " [
        SS9   U" [        R                  " U SSSSS95        S S S 5        g ! , (       d  f       g = f)	N)   r   z+Error in determining null-space with ARPACK)matchr   r   rQ   r   )r0   r1   rO   r5   )r   r   r   pytestraises
ValueErrorr   locally_linear_embedding)Mfs     r#   test_singular_matrixrp      sV    
AA	z)V	W	--!%	
 
X	W	Ws   A
A,c                      [         R                  R                  S5      n U R                  SSSS9nS H)  n[        R
                  " USS9nUR                  U5        M+     g )Nr   r7   )   r   r,   rP   
   )rO   r0   )r   r8   r9   randintr   r>   rA   )randr   rO   rG   s       r#   test_integer_inputrv      sQ    99  #DQ'*A=--VL
 >r%   c                      [        SSS9u  pSn[        R                  " US9nUR                  U 5        UR	                  5       n[        [        U5       Vs/ s H  nSU 3PM
     snU5        gs  snf )z7Check get_feature_names_out for LocallyLinearEmbedding.r      )r2   
n_featuresr   )r1   locallylinearembeddingN)r   r   r>   rA   get_feature_names_outr
   r;   )r   re   r1   isonamesis         r#   test_get_feature_names_outr      sl    13DAL

)
)|
DCGGAJ%%'E/4\/BC/B!!!	%/BCUCs   A.)	itertoolsr   numpyr   rj   scipyr   ra   r   r   sklearn.datasetsr    sklearn.manifold._locally_linearr   sklearn.utils._testingr	   r
   r   r?   r$   rN   markparametrizer\   rf   rp   rv   r    r%   r#   <module>r      s        ' ' H  (#
2.%<P #NO=1 2 P@!$
 
r%   