source: orange/orange/OrangeWidgets/Unsupervised/OWNxClustering.py @ 9622:3f0d63a65934

Revision 9622:3f0d63a65934, 2.8 KB checked in by Miha Stajdohar <miha.stajdohar@…>, 2 years ago (diff)

Fixed signal names.

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
10import Orange.network.community as cd
11import OWGUI
12
13from OWWidget import *
14
15
16class OWNxClustering(OWWidget):
17
18    settingsList = ['method', 'iterationHistory', 'autoApply', 'iterations', 'hop_attenuation']
19
20    def __init__(self, parent=None, signalManager=None):
21        OWWidget.__init__(self, parent, signalManager, 'Nx Clustering')
22
23        self.inputs = [("Network", Orange.network.Graph, self.setNetwork, Default)]
24        self.outputs = [("Network", Orange.network.Graph)]
25
26        self.net = None
27        self.method = 0
28        self.iterationHistory = 0
29        self.autoApply = 0
30        self.iterations = 1000
31        self.hop_attenuation = 0.1
32        self.loadSettings()
33
34        OWGUI.spin(self.controlArea, self, "iterations", 1, 100000, 1, label="Iterations: ")
35        ribg = OWGUI.radioButtonsInBox(self.controlArea, self, "method", [], "Method", callback=self.cluster)
36        OWGUI.appendRadioButton(ribg, self, "method", "Label propagation clustering (Raghavan et al., 2007)", callback=self.cluster)
37
38        OWGUI.appendRadioButton(ribg, self, "method", "Label propagation clustering (Leung et al., 2009)", callback=self.cluster)
39        OWGUI.doubleSpin(OWGUI.indentedBox(ribg), self, "hop_attenuation", 0, 1, 0.01, label="Hop attenuation (delta): ")
40
41
42        self.info = OWGUI.widgetLabel(self.controlArea, ' ')
43        OWGUI.checkBox(self.controlArea, self, "iterationHistory", "Append clustering data on each iteration")
44        autoApplyCB = OWGUI.checkBox(self.controlArea, self, "autoApply", "Commit automatically")
45        OWGUI.button(self.controlArea, self, "Commit", callback=self.cluster)
46
47    def setNetwork(self, net):
48        self.net = net
49        if self.autoApply:
50            self.cluster()
51
52    def cluster(self):
53        self.info.setText(' ')
54
55        if self.net is None:
56            self.send("Network", None)
57            return
58
59        if self.method == 0:
60            labels = cd.label_propagation(self.net, results2items=1,
61                                    resultHistory2items=self.iterationHistory,
62                                    iterations=self.iterations)
63        if self.method == 1:
64            labels = cd.label_propagation_hop_attenuation(
65                                    self.net,
66                                    results2items=1,
67                                    resultHistory2items=self.iterationHistory,
68                                    iterations=self.iterations,
69                                    delta=self.hop_attenuation)
70
71        self.info.setText('%d clusters found' % len(set(labels.items())))
72        self.send("Network", self.net)
Note: See TracBrowser for help on using the repository browser.