source: orange/orange/OrangeWidgets/Regression/OWMean.py @ 9315:e073f5001ad5

Revision 9315:e073f5001ad5, 1.9 KB checked in by ales_erjavec <ales.erjavec@…>, 2 years ago (diff)

Set predictor.name in SVM Regression.
Override setData in Mean (to check for continuous class variable)

Line 
1"""
2<name>Mean</name>
3<description>Mean regression</description>
4<icon>icons/Mean.png</icon>
5<priority>5</priority>
6<contact>Ales Erjavec (ales.erjavec(@at@)fri.uni-lj.si)</contact>
7<keywords>mean, average</keywords>
8
9"""
10
11from OWMajority import *
12
13class OWMean(OWMajority):
14    def __init__(self, parent=None, signalManager=None, title="Mean"):
15        OWWidget.__init__(self, parent, signalManager, title, wantMainArea=False)
16
17        self.inputs = [("Examples", ExampleTable, self.setData),
18                       ("Preprocess", PreprocessedLearner, self.setPreprocessor)]
19       
20        self.outputs = [("Learner", orange.Learner),
21                        ("Predictor", orange.Classifier)]
22       
23       
24        self.name = 'Mean'
25       
26        self.loadSettings()
27       
28        self.data = None
29        self.preprocessor = None
30
31        OWGUI.lineEdit(self.controlArea, self, 'name', 
32                       box='Learner/Predictor Name', \
33                       tooltip='Name to be used by other widgets to identify your learner/predictor.')
34
35        OWGUI.separator(self.controlArea)
36
37        OWGUI.button(self.controlArea, self, "&Apply", 
38                     callback=self.setLearner,
39                     disabled=0,
40                     default=True)
41       
42        OWGUI.rubber(self.controlArea)
43       
44        self.learner = orange.MajorityLearner()
45        self.setLearner()
46        self.resize(100,100)
47       
48    def setData(self,data):
49        self.data = self.isDataWithClass(data, orange.VarTypes.Continuous, checkMissing=True) and data or None
50
51        if self.data:
52            try:
53                self.classifier = self.learner(self.data)
54                self.classifier.name = self.name
55                self.error(1)
56            except Exception, (errValue):
57                self.classifier = None
58                self.error(1, str(errValue))
59        else:
60            self.classifier = None
61        self.send("Predictor", self.classifier)
62       
Note: See TracBrowser for help on using the repository browser.