35808c2edf30be03ad2bec8264f64ef327fab050,methods/scikit/linear_ridge_regression.py,LinearRidgeRegression,RunMetrics,#LinearRidgeRegression#Any#,118

Before Change

      return metrics_dict

    else:
      Log.Fatal("This method requires three datasets.")

After Change

  def RunMetrics(self, options):
    Log.Info("Perform Linear Ridge Regression.", self.verbose)

    results = self.LinearRidgeRegressionScikit(options)
    if results < 0:
      return results

    // Datastructure to store the results.
    metrics = {"Runtime" : results}

    if len(self.dataset) >= 3:

      trainData, labels = SplitTrainData(self.dataset)

      testData = LoadDataset(self.dataset[1])
      truelabels = LoadDataset(self.dataset[2])
      alpha = re.search("-t (\d+)", options)
      alpha = 1.0 if not alpha else int(alpha.group(1))

      predictedlabels = np.rint(self.BuildModel(trainData, labels, alpha=alpha).predict(testData))

      SimpleMSE = Metrics.SimpleMeanSquaredError(truelabels, predictedlabels)
      metrics["Simple MSE"] = SimpleMSE

    return metrics

Instances