Ejemplo n.º 1
0
 def test_regression(self):
     print('testing regression metric')
     regression_metric.R2Score().compute(self.reg_score, self.reg_label)
     regression_metric.MSE().compute(self.reg_score, self.reg_label)
     regression_metric.RMSE().compute(self.reg_score, self.reg_label)
     regression_metric.ExplainedVariance().compute(self.reg_score, self.reg_label)
     regression_metric.Describe().compute(self.reg_score)
Ejemplo n.º 2
0
 def r2_score(labels, pred_scores):
     """
     Compute R^2 (coefficient of determination) score
     Parameters
     ----------
     labels: value list. The labels of data set.
     pred_scores: value list. The predict results of model. It should be corresponding to labels each data.
     Returns
     ----------
     float
         The R^2 score
     """
     return regression_metric.R2Score().compute(labels, pred_scores)