Changeset 9314:945c3af834ee in orange


Ignore:
Timestamp:
12/06/11 13:42:39 (2 years ago)
Author:
ales_erjavec <ales.erjavec@…>
Branch:
default
Convert:
3c1e2b52bd74104fc94a7cd0056dfc30a608cc48
Message:

Overload setData method (for class type check).

File:
1 edited

Legend:

Unmodified
Added
Removed
  • orange/OrangeWidgets/Regression/OWRandomForestRegression.py

    r9223 r9314  
    1515        OWRandomForest.__init__(self, parent, signalManager, title) 
    1616         
    17         self.inputs = [("Examples", ExampleTable, self.setData), ("Preprocess", PreprocessedLearner, self.setPreprocessor)] 
    18         self.outputs = [("Learner", orange.Learner),("Random Forest Classifier", orange.Classifier),("Choosen Tree", orange.TreeClassifier) ] 
     17        self.inputs = [("Examples", ExampleTable, self.setData), 
     18                       ("Preprocess", PreprocessedLearner, self.setPreprocessor)] 
    1919         
     20        self.outputs = [("Learner", orange.Learner), 
     21                        ("Random Forest Classifier", orange.Classifier), 
     22                        ("Choosen Tree", orange.TreeClassifier)] 
     23         
     24 
     25    def setData(self, data): 
     26        self.data = self.isDataWithClass(data, orange.VarTypes.Continuous, checkMissing=True) and data or None 
     27         
     28        if self.data: 
     29            learner = self.constructLearner() 
     30            pb = OWGUI.ProgressBar(self, iterations=self.trees) 
     31            learner.callback = pb.advance 
     32            try: 
     33                self.classifier = learner(self.data) 
     34                self.classifier.name = self.name 
     35                self.streeEnabled(True) 
     36            except Exception, (errValue): 
     37                self.error(str(errValue)) 
     38                self.classifier = None 
     39                self.streeEnabled(False) 
     40            pb.finish() 
     41        else: 
     42            self.classifier = None 
     43            self.streeEnabled(False) 
     44 
     45        self.send("Random Forest Classifier", self.classifier) 
     46         
Note: See TracChangeset for help on using the changeset viewer.