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
|
from Orange.base import Learner, Model, SklLearner, SklModel
__all__ = ["LearnerRegression", "ModelRegression",
"SklModelRegression", "SklLearnerRegression"]
class LearnerRegression(Learner):
def incompatibility_reason(self, domain):
reason = None
if len(domain.class_vars) > 1 and not self.supports_multiclass:
reason = "Too many target variables."
elif not domain.has_continuous_class:
reason = "Numeric target variable expected."
return reason
class ModelRegression(Model):
pass
class SklModelRegression(SklModel, ModelRegression):
pass
class SklLearnerRegression(SklLearner, LearnerRegression):
__returns__ = SklModelRegression
|