\Mh.JdZddlmZddlZddlmZddlmZddlm Z ddgZ Gd d eZ d Z e e d Z ejd ZejdZejdZejdddZe ejdZdS)zHFunctions for measuring the quality of a partition (into communities). ) combinationsN) NetworkXError) is_partition)argmap modularitypartition_qualityc"eZdZdZfdZxZS) NotAPartitionz0Raised if a given collection is not a partition.cX|d|}t|dS)Nz' is not a valid partition of the graph )super__init__)selfG collectionmsg __class__s e/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/networkx/algorithms/community/quality.pyr zNotAPartition.__init__s4GGAGG )__name__ __module__ __qualname____doc__r __classcell__)rs@rr r s>::rr cRt||r||fStjd)aDecorator to check that a valid partition is input to a function Raises :exc:`networkx.NetworkXError` if the partition is not valid. This decorator should be used on functions whose first two arguments are a graph and a partition of the nodes of that graph (in that order):: >>> @require_partition ... def foo(G, partition): ... print("partition is valid!") ... >>> G = nx.complete_graph(5) >>> partition = [{0, 1}, {2, 3}, {4}] >>> foo(G, partition) partition is valid! >>> partition = [{0}, {2, 3}, {4}] >>> foo(G, partition) Traceback (most recent call last): ... networkx.exception.NetworkXError: `partition` is not a valid partition of the nodes of G >>> partition = [{0, 1}, {1, 2, 3}, {4}] >>> foo(G, partition) Traceback (most recent call last): ... networkx.exception.NetworkXError: `partition` is not a valid partition of the nodes of G z6`partition` is not a valid partition of the nodes of G)rnxrr partitions r_require_partitionrs2:Ay!!)| S T TTr)rc:tfd|DS)aRReturns the number of intra-community edges for a partition of `G`. Parameters ---------- G : NetworkX graph. partition : iterable of sets of nodes This must be a partition of the nodes of `G`. The "intra-community edges" are those edges joining a pair of nodes in the same block of the partition. c3fK|]+}|V,dSN)subgraphsize).0blockrs r z(intra_community_edges..Ls;??Eqzz%  %%''??????r)sumrs` rintra_community_edgesr)=s( ????Y??? ? ??rc|r tjn tj}tj|||S)aReturns the number of inter-community edges for a partition of `G`. according to the given partition of the nodes of `G`. Parameters ---------- G : NetworkX graph. partition : iterable of sets of nodes This must be a partition of the nodes of `G`. The *inter-community edges* are those edges joining a pair of nodes in different blocks of the partition. Implementation note: this function creates an intermediate graph that may require the same amount of memory as that of `G`. ) create_using) is_directedr MultiDiGraph MultiGraphquotient_graphr$)rrMGs rinter_community_edgesr1OsB8MMOO >B Q  ; ; ; @ @ B BBrcFttj||S)aReturns the number of inter-community non-edges according to the given partition of the nodes of `G`. Parameters ---------- G : NetworkX graph. partition : iterable of sets of nodes This must be a partition of the nodes of `G`. A *non-edge* is a pair of nodes (undirected if `G` is undirected) that are not adjacent in `G`. The *inter-community non-edges* are those non-edges on a pair of nodes in different blocks of the partition. Implementation note: this function creates two intermediate graphs, which may require up to twice the amount of memory as required to store `G`. )r1r complementrs rinter_community_non_edgesr4os< !q!1!19 = ==rweight) edge_attrsrc t|tst|}t|st|rpt  t t ddzz nSt x t }|dz d|dzz  fd}tt||S)a7 Returns the modularity of the given partition of the graph. Modularity is defined in [1]_ as .. math:: Q = \frac{1}{2m} \sum_{ij} \left( A_{ij} - \gamma\frac{k_ik_j}{2m}\right) \delta(c_i,c_j) where $m$ is the number of edges (or sum of all edge weights as in [5]_), $A$ is the adjacency matrix of `G`, $k_i$ is the (weighted) degree of $i$, $\gamma$ is the resolution parameter, and $\delta(c_i, c_j)$ is 1 if $i$ and $j$ are in the same community else 0. According to [2]_ (and verified by some algebra) this can be reduced to .. math:: Q = \sum_{c=1}^{n} \left[ \frac{L_c}{m} - \gamma\left( \frac{k_c}{2m} \right) ^2 \right] where the sum iterates over all communities $c$, $m$ is the number of edges, $L_c$ is the number of intra-community links for community $c$, $k_c$ is the sum of degrees of the nodes in community $c$, and $\gamma$ is the resolution parameter. The resolution parameter sets an arbitrary tradeoff between intra-group edges and inter-group edges. More complex grouping patterns can be discovered by analyzing the same network with multiple values of gamma and then combining the results [3]_. That said, it is very common to simply use gamma=1. More on the choice of gamma is in [4]_. The second formula is the one actually used in calculation of the modularity. For directed graphs the second formula replaces $k_c$ with $k^{in}_c k^{out}_c$. Parameters ---------- G : NetworkX Graph communities : list or iterable of set of nodes These node sets must represent a partition of G's nodes. weight : string or None, optional (default="weight") The edge attribute that holds the numerical value used as a weight. If None or an edge does not have that attribute, then that edge has weight 1. resolution : float (default=1) If resolution is less than 1, modularity favors larger communities. Greater than 1 favors smaller communities. Returns ------- Q : float The modularity of the partition. Raises ------ NotAPartition If `communities` is not a partition of the nodes of `G`. Examples -------- >>> G = nx.barbell_graph(3, 0) >>> nx.community.modularity(G, [{0, 1, 2}, {3, 4, 5}]) 0.35714285714285715 >>> nx.community.modularity(G, nx.community.label_propagation_communities(G)) 0.35714285714285715 References ---------- .. [1] M. E. J. Newman "Networks: An Introduction", page 224. Oxford University Press, 2011. .. [2] Clauset, Aaron, Mark EJ Newman, and Cristopher Moore. "Finding community structure in very large networks." Phys. Rev. E 70.6 (2004). .. [3] Reichardt and Bornholdt "Statistical Mechanics of Community Detection" Phys. Rev. E 74, 016110, 2006. https://doi.org/10.1103/PhysRevE.74.016110 .. [4] M. E. J. Newman, "Equivalence between modularity optimization and maximum likelihood methods for community detection" Phys. Rev. E 94, 052315, 2016. https://doi.org/10.1103/PhysRevE.94.052315 .. [5] Blondel, V.D. et al. "Fast unfolding of communities in large networks" J. Stat. Mech 10008, 1-12 (2008). https://doi.org/10.1088/1742-5468/2008/10/P10008 )r5rct|tfd dD}t fdD}rtfdDn|}|z |z|z zz S)Nc3,K|]\}}}|v |VdSr")r%uvwtcomms rr'z=modularity..community_contribution..s.XXAraSWii"iiiiXXrr)datadefaultc3(K|] }|V dSr"r;)r%r< out_degrees rr'z=modularity..community_contribution..s'99qZ]999999rc3(K|] }|V dSr"r;)r%r< in_degrees rr'z=modularity..community_contribution..s'77QIaL777777r)setr(edges) communityL_cout_degree_sum in_degree_sumr?rdirectedrEmnormrC resolutionr5s @rcommunity_contributionz*modularity..community_contributions9~~XXXXQWWTW%J%JXXXXX9999D99999;CW7777$777777 Qwn4}DtKKKr) isinstancelistrr r,dictrCrEr(valuesdegreemap) r communitiesr5rOdeg_sumrPrLrErMrNrCs ` `` @@@@@rrrssj k4 ( ((;'' ; ' ',A{+++}}H !,,f,5566 F3344  !!## $ $1a4x!%ahhfh&=&=!>!>> Yj''))** aK7A:~LLLLLLLLLLLL s);77 8 88rcNi}t|D]\}}|D]}|||<|sAtdt|dD}|r|dz}nd}t |}||dz z}|s|dz}d} |} |D]+} || d|| dkr| dz } &| dz} ,| t |jz } |rd} n| | z|z } | | fS)aWReturns the coverage and performance of a partition of G. The *coverage* of a partition is the ratio of the number of intra-community edges to the total number of edges in the graph. The *performance* of a partition is the number of intra-community edges plus inter-community non-edges divided by the total number of potential edges. This algorithm has complexity $O(C^2 + L)$ where C is the number of communities and L is the number of links. Parameters ---------- G : NetworkX graph partition : sequence Partition of the nodes of `G`, represented as a sequence of sets of nodes (blocks). Each block of the partition represents a community. Returns ------- (float, float) The (coverage, performance) tuple of the partition, as defined above. Raises ------ NetworkXError If `partition` is not a valid partition of the nodes of `G`. Notes ----- If `G` is a multigraph; - for coverage, the multiplicity of edges is counted - for performance, the result is -1 (total number of possible edges is not defined) References ---------- .. [1] Santo Fortunato. "Community Detection in Graphs". *Physical Reports*, Volume 486, Issue 3--5 pp. 75--174 c3ZK|]&\}}t|t|zV'dSr")len)r%p1p2s rr'z$partition_quality..:sH- - "("bCGGc"gg - - - - - - rr8rrg) enumerate is_multigraphr(rr,r[rG)rrnode_communityirHnodepossible_inter_community_edgesn total_pairsr)r4ecoverage performances rrrs`N!),,%% 9 % %D#$N4  % ??   +),- - ,8A,F,F- - - * * & ==?? 0 *a / *)*& AAq1u+K ==??  >WWYY++ !A$ >!A$#7 7 7 !Q & ! ! % * % %$s17||3HX ,/HHKW [  r)r5r)r itertoolsrnetworkxrr-networkx.algorithms.community.community_utilsrnetworkx.utils.decoratorsr__all__r rrequire_partition _dispatchabler)r1r4rrr;rrrps #"""""""""""FFFFFF,,,,,, , -MUUUDF-v66@@@"CCC>>>>@X&&&n9n9n9'&n9bW!W!W!W!W!r