Ë
    çÍ:jw  ã                   ón   — 	 d dl Z d dlmZmZmZ  G d„ de«      Zd„ Zedk(  r e«        yy# e$ r Y Œ-w xY w)é    N)Ú
DendrogramÚVectorSpaceClustererÚcosine_distancec                   óL   — e Zd ZdZdd„Zdd„Zdd„Zd„ Zd„ Zd„ Z	d	„ Z
d
„ Zd„ Zy)ÚGAAClustereraM  
    The Group Average Agglomerative starts with each of the N vectors as singleton
    clusters. It then iteratively merges pairs of clusters which have the
    closest centroids.  This continues until there is only one cluster. The
    order of merges gives rise to a dendrogram: a tree with the earlier merges
    lower than later merges. The membership of a given number of clusters c, 1
    <= c <= N, can be found by cutting the dendrogram at depth c.

    This clusterer uses the cosine similarity metric only, which allows for
    efficient speed-up in the clustering process.
    Nc                 ó\   — t        j                  | ||«       || _        d | _        d | _        y ©N)r   Ú__init__Ú_num_clustersÚ_dendrogramÚ_groups_values)ÚselfÚnum_clustersÚ	normaliseÚsvd_dimensionss       úf/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/nltk/cluster/gaac.pyr
   zGAAClusterer.__init__   s,   € Ü×%Ñ% d¨I°~ÔFØ)ˆÔØˆÔØ"ˆÕó    c           	      ó¼   — t        |D �cg c]&  }t        j                  |t        j                  «      ‘Œ( c}«      | _        t        j                  | |||«      S c c}w r	   )r   ÚnumpyÚarrayÚfloat64r   r   Úcluster)r   ÚvectorsÚassign_clustersÚtraceÚvectors        r   r   zGAAClusterer.cluster#   sK   € ä%Ø>EÖF°FŒU�[‰[˜¤§¡Õ/ÒFó
ˆÔô $×+Ñ+¨D°'¸?ÈEÓRÐRùò Gs   Š+Ac                 óR  — t        |«      }dg|z  }|}t        j                  |«      }||f}t        j                  |t        ¬«      t        j
                  z  }t        |«      D ]-  }	t        |	dz   |«      D ]  }
t        ||	   ||
   «      ||	|
f<   Œ Œ/ |t        | j                  d«      kD  ràt        j                  |j                  «       |«      \  }	}
|rt        d|	|
fz  «       | j                  |||	|
«       t        j
                  |d d …|
f<   t        j
                  ||
d d …f<   ||	   ||
   z   ||	<   | j                  j                  ||	   ||
   «       |dz  }||
dz   d xxx dz  ccc |||
<   |t        | j                  d«      kD  rŒà| j!                  | j                  «       y )Né   )Údtypezmerging %d and %d)Úlenr   ÚarangeÚonesÚfloatÚinfÚranger   Úmaxr   Úunravel_indexÚargminÚprintÚ_merge_similaritiesr   ÚmergeÚupdate_clusters)r   r   r   ÚNÚcluster_lenÚcluster_countÚ	index_mapÚdimsÚdistÚiÚjs              r   Úcluster_vectorspacez GAAClusterer.cluster_vectorspace*   s¨  € ä�‹LˆØ�c˜A‘gˆØˆÜ—L‘L “Oˆ	ð �1ˆvˆÜ�z‰z˜$¤eÔ,¬u¯y©yÑ8ˆÜ�q“ò 	EˆAÜ˜1˜q™5 !“_ò E�Ü,¨W°Q©Z¸À¹ÓD��Q˜�T’
ñEð	Eð œc $×"4Ñ"4°aÓ8Ò8Ü×&Ñ& t§{¡{£}°dÓ;‰DˆAˆqÙÜÐ)¨Q°¨FÑ2Ô3ð ×$Ñ$ T¨;¸¸1Ô=ô Ÿ™ˆD’�A�‰JÜŸ™ˆD�’A�‰Jð )¨™^¨k¸!©nÑ<ˆK˜‰NØ×Ñ×"Ñ" 9¨Q¡<°¸1±Ô>Ø˜QÑˆMð �a˜!‘e�gÓ !Ñ#ÓØˆI�a‰Lð) œc $×"4Ñ"4°aÓ8Ó8ð, 	×Ñ˜T×/Ñ/Õ0r   c                 óP  — ||   }||   }||z   }|d |…|f   |z  |d |…|f   |z  z   |d |…|f<   |d |…|fxx   |z  cc<   |||dz   |…f   |z  ||dz   |…|f   |z  z   |||dz   |…f<   |||dz   d …f   |z  |||dz   d …f   |z  z   |||dz   d …f<   |||dz   d …fxx   |z  cc<   y )Nr   © )r   r2   r.   r3   r4   Úi_weightÚj_weightÚ
weight_sums           r   r*   z GAAClusterer._merge_similaritiesP   s  € ð ˜q‘>ˆØ˜q‘>ˆØ Ñ(ˆ
ð ˜2˜A˜2˜q˜5‘k HÑ,¨t°B°Q°B¸°E©{¸XÑ/EÑEˆˆRˆaˆR�ˆU‰ØˆRˆaˆR�ˆU‹�zÑ!‹ð ��A˜‘E˜A�I�Ñ Ñ)¨D°°Q±¸°¸A°Ñ,>ÀÑ,IÑIð 	ˆQ��A‘˜�	ˆ\Ñð    1 q¡5¡7 
Ñ+¨hÑ6¸¸aÀÀQÁÁ¸jÑ9IÈHÑ9TÑTˆˆQ��A‘‘ˆZÑØˆQ��A‘‘ˆZÓ˜JÑ&Ôr   c                 óÎ  — | j                   j                  |«      }g | _        |D ]¤  }t        |«      dkD  sJ ‚| j                  r| j                  |d   «      }nt        j                  |d   «      }|dd  D ](  }| j                  r|| j                  |«      z  }Œ$||z  }Œ* |t        |«      z  }| j                  j                  |«       Œ¦ t        | j                  «      | _	        y ©Nr   r   )
r   ÚgroupsÚ
_centroidsr    Ú_should_normaliseÚ
_normaliser   r   Úappendr   )r   r   Úclustersr   Úcentroidr   s         r   r,   zGAAClusterer.update_clustersc   sÝ   € Ø×#Ñ#×*Ñ*¨<Ó8ˆØˆŒØò 	-ˆGÜ�w“< !Ò#Ð#Ð#Ø×%Ò%ØŸ?™?¨7°1©:Ó6‘ä Ÿ;™; w¨q¡zÓ2�Ø! ! "˜+ò '�Ø×)Ò)Ø §¡°Ó 7Ñ7‘Hà Ñ&‘Hð	'ð
 œ˜G›Ñ$ˆHØ�O‰O×"Ñ" 8Õ,ð	-ô ! §¡Ó1ˆÕr   c                 ó˜   — d }t        | j                  «      D ],  }| j                  |   }t        ||«      }|r	||d   k  sŒ)||f}Œ. |d   S r<   )r%   r   r>   r   )r   r   Úbestr3   rC   r2   s         r   Úclassify_vectorspacez!GAAClusterer.classify_vectorspaceu   s\   € ØˆÜ�t×)Ñ)Ó*ò 	!ˆAØ—‘ qÑ)ˆHÜ" 6¨8Ó4ˆDÙ˜4 $ q¡'›>Ø˜a�y‘ð		!ð
 �A‰wˆr   c                 ó   — | j                   S )zi
        :return: The dendrogram representing the current clustering
        :rtype:  Dendrogram
        )r   ©r   s    r   Ú
dendrogramzGAAClusterer.dendrogram~   s   € ð
 ×ÑÐr   c                 ó   — | j                   S r	   ©r   rH   s    r   r   zGAAClusterer.num_clusters…   s   € Ø×!Ñ!Ð!r   c                 ó    — d| j                   z  S )Nz*<GroupAverageAgglomerative Clusterer n=%d>rK   rH   s    r   Ú__repr__zGAAClusterer.__repr__ˆ   s   € Ø;¸d×>PÑ>PÑPÐPr   )r   TN)FF)F)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r
   r   r5   r*   r,   rF   rI   r   rM   r7   r   r   r   r      s7   „ ñ
ó#óSó$1òL'ò&2ò$ò ò"óQr   r   c                  óä  — ddl m}  ddgddgddgddgddgddgfD �cg c]  }t        j                  |«      ‘Œ }} | d«      }|j	                  |d«      }t        d|«       t        d	|«       t        d
|«       t        «        |j                  «       j                  «        t        j                  ddg«      }t        d|z  d¬«       t        |j                  |«      «       t        «        yc c}w )zO
    Non-interactive demonstration of the clusterers with simple 2-D data.
    r   )r   é   r   é   é   Tz
Clusterer:z
Clustered:zAs:zclassify(%s):ú )ÚendN)	Únltk.clusterr   r   r   r   r)   rI   ÚshowÚclassify)r   Úfr   Ú	clustererrB   r   s         r   Údemor]   Œ   sä   € õ
 *ð *+¨A¨°°A°¸¸A¸ÀÀAÀÈÈAÈÐQRÐTUÐPVÐ'WÖX !Œu�{‰{˜1�~ÐX€GÐXñ ˜Q“€IØ× Ñ  ¨$Ó/€Hä	ˆ,˜	Ô"Ü	ˆ,˜Ô Ü	ˆ%�ÔÜ	„Gð ×ÑÓ×ÑÔ!ô �[‰[˜!˜Q˜Ó €FÜ	ˆ/˜FÑ
"¨Õ,Ü	ˆ)×
Ñ
˜VÓ
$Ô%Ü	…Gùò% Ys   �C-Ú__main__)	r   ÚImportErrorÚnltk.cluster.utilr   r   r   r   r]   rN   r7   r   r   ú<module>ra      sV   ðð	Û÷ PÑ OôyQÐ'ô yQòxð: ˆzÒÙ…Fð øð ò 	Ùð	ús   ‚, ¬4³4