source: orange/orange/OrangeWidgets/Unsupervised/OWNxClustering.py @ 9630:112c812933ee

Revision 9630:112c812933ee, 3.2 KB checked in by Miha Stajdohar <miha.stajdohar@…>, 2 years ago (diff)

Fixed a bug when appending clustering results multiple times.

Line 
1"""
2<name>Net Clustering</name>
3<description>Orange widget for community detection in networks</description>
4<icon>icons/Network.png</icon>
5<contact>Miha Stajdohar (miha.stajdohar(@at@)gmail.com)</contact>
6<priority>6430</priority>
7"""
8
9import Orange.network
10import Orange.network.community as cd
11import OWGUI
12
13from OWWidget import *
14
15
16class OWNxClustering(OWWidget):
17
18    settingsList = ['method', 'iterationHistory', 'autoApply', 'iterations',
19                    'hop_attenuation']
20
21    def __init__(self, parent=None, signalManager=None):
22        OWWidget.__init__(self, parent, signalManager, 'Nx Clustering')
23
24        self.inputs = [("Network", Orange.network.Graph,
25                        self.setNetwork, Default)]
26        self.outputs = [("Network", Orange.network.Graph)]
27
28        self.net = None
29        self.method = 0
30        self.iterationHistory = 0
31        self.autoApply = 0
32        self.iterations = 1000
33        self.hop_attenuation = 0.1
34        self.loadSettings()
35
36        OWGUI.spin(self.controlArea, self, "iterations", 1,
37                   100000, 1, label="Iterations: ")
38        ribg = OWGUI.radioButtonsInBox(self.controlArea, self, "method",
39                                       [], "Method", callback=self.cluster)
40        OWGUI.appendRadioButton(ribg, self, "method",
41                        "Label propagation clustering (Raghavan et al., 2007)",
42                        callback=self.cluster)
43
44        OWGUI.appendRadioButton(ribg, self, "method",
45                        "Label propagation clustering (Leung et al., 2009)",
46                        callback=self.cluster)
47        OWGUI.doubleSpin(OWGUI.indentedBox(ribg), self, "hop_attenuation",
48                         0, 1, 0.01, label="Hop attenuation (delta): ")
49
50        self.info = OWGUI.widgetLabel(self.controlArea, ' ')
51        OWGUI.checkBox(self.controlArea, self, "iterationHistory",
52                       "Append clustering data on each iteration",
53                       callback=self.cluster)
54        OWGUI.checkBox(self.controlArea, self, "autoApply",
55                       "Commit automatically")
56        OWGUI.button(self.controlArea, self, "Commit",
57                     callback=lambda b=True: self.cluster(b))
58
59    def setNetwork(self, net):
60        self.net = net
61        if self.autoApply:
62            self.cluster()
63
64    def cluster(self, btn=False):
65        if not btn and not self.autoApply:
66            return
67
68        self.info.setText(' ')
69
70        if self.net is None:
71            self.send("Network", None)
72            return
73
74        if self.method == 0:
75            labels = cd.label_propagation(self.net, results2items=1,
76                                    resultHistory2items=self.iterationHistory,
77                                    iterations=self.iterations)
78        if self.method == 1:
79            labels = cd.label_propagation_hop_attenuation(
80                                    self.net,
81                                    results2items=1,
82                                    resultHistory2items=self.iterationHistory,
83                                    iterations=self.iterations,
84                                    delta=self.hop_attenuation)
85
86        self.info.setText('%d clusters found' % len(set(labels.values())))
87        self.send("Network", self.net)
Note: See TracBrowser for help on using the repository browser.