added is_multi_resp attribute

Author: idc9

ya_glm/models/huber_regression.py

@@ -4,6 +4,7 @@
 
 
 class HuberRegMixin(LinRegMixin):
+    is_multi_resp = False
 
     @autoassign
     def __init__(self, knot=1.35): pass
@@ -16,6 +17,8 @@ def get_loss_info(self):
 
 
 class HuberRegMultiResponseMixin(LinRegMultiResponseMixin):
+    is_multi_resp = True
+
     @autoassign
     def __init__(self, knot=1.35): pass
 

ya_glm/models/linear_regression.py

@@ -10,6 +10,9 @@
 
 class LinRegMixin(RegressorMixin):
 
+    is_multi_resp = False
+
+
     def get_loss_info(self):
         loss_type = 'lin_reg'
         loss_kws = {}

ya_glm/models/linear_regression_multi_resp.py

@@ -5,6 +5,8 @@
 
 class LinRegMultiResponseMixin(LinRegMixin):
 
+    is_multi_resp = True
+
     def get_loss_info(self):
         loss_type = 'lin_reg_mr'
         loss_kws = {}

ya_glm/models/logistic_regression.py

@@ -15,6 +15,8 @@
 
 class LogRegMixin(LinearClassifierMixin):
 
+    is_multi_resp = False
+
     def get_loss_info(self):
         loss_type = 'log_reg'
         loss_kws = {}

ya_glm/models/multinomial.py

@@ -11,6 +11,8 @@
 
 class MultinomialMixin(LinearClassifierMixin):
 
+    is_multi_resp = True
+
     def get_loss_info(self):
         loss_type = 'multinomial'
         loss_kws = {}

ya_glm/models/poisson_regression.py

@@ -7,6 +7,7 @@
 
 
 class PoissonRegMixin(RegressorMixin):
+    is_multi_resp = False
 
     def get_loss_info(self):
         loss_type = 'poisson'
@@ -41,6 +42,8 @@ def score(self, X, y, sample_weight=None):
 
 class PoissonRegMultiResponseMixin(RegressorMixin):
 
+    is_multi_resp = True
+
     def get_loss_info(self):
         loss_type = 'poisson_mr'
         loss_kws = {}

ya_glm/models/quantile_regression.py

@@ -6,6 +6,8 @@
 
 class QuantileRegMixin(LinRegMixin):
 
+    is_multi_resp = False
+
     @autoassign
     def __init__(self, quantile=0.5): pass