-
Notifications
You must be signed in to change notification settings - Fork 0
/
new_examples.py
121 lines (70 loc) · 2.23 KB
/
new_examples.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
"""Example runs with Karate Club."""
import networkx as nx
from karateclub.community_detection.overlapping import EgoNetSplitter, NNSED, DANMF, MNMF
from karateclub.community_detection.non_overlapping import EdMot, LabelPropagation
from karateclub.node_embedding.neighbourhood import GraRep
from karateclub.node_embedding.structural import GraphWave
#------------------------------------
# Splitter example
#------------------------------------
g = nx.newman_watts_strogatz_graph(100, 20, 0.05)
model = EgoNetSplitter(1.0)
model.fit(g)
print(model.get_memberships())
#------------------------------------
# Edmot example
#------------------------------------
g = nx.newman_watts_strogatz_graph(100, 10, 0.9)
model = EdMot(3, 0.5)
model.fit(g)
print(model.get_memberships())
#------------------------------------
# DANMF example
#------------------------------------
g = nx.newman_watts_strogatz_graph(100, 10, 0.02)
model = DANMF()
model.fit(g)
print(model.get_memberships())
print(model.get_embedding())
#------------------------------------
# M-NMF example
#------------------------------------
g = nx.newman_watts_strogatz_graph(100, 10, 0.02)
model = MNMF()
model.fit(g)
print(model.get_memberships())
print(model.get_embedding())
print(model.get_cluster_centers())
#------------------------------------
# Label Propagation example
#------------------------------------
g = nx.newman_watts_strogatz_graph(100, 10, 0.02)
model = LabelPropagation()
model.fit(g)
print(model.get_memberships())
#------------------------------------
# GraRep example
#------------------------------------
g = nx.newman_watts_strogatz_graph(100, 10, 0.02)
model = GraRep()
model.fit(g)
embedding = model.get_embedding()
print(embedding)
#------------------------------------
# GraphWave example
#------------------------------------
g = nx.newman_watts_strogatz_graph(100, 10, 0.02)
model = GraphWave()
model.fit(g)
embedding = model.get_embedding()
print(embedding)
#------------------------------------
# NNSED example
#------------------------------------
g = nx.newman_watts_strogatz_graph(100, 10, 0.02)
model = NNSED()
model.fit(g)
embedding = model.get_embedding()
print(embedding)
memberships = model.get_memberships()
print(memberships)