/** * Average MSE across all columns * @return */ public double averageMeanSquaredError() { double ret = 0.0; for (int i = 0; i < numColumns(); i++) { ret += meanSquaredError(i); } return ret / (double) numColumns(); }
/** * Average MAE across all columns * @return */ public double averageMeanAbsoluteError() { double ret = 0.0; for (int i = 0; i < numColumns(); i++) { ret += meanAbsoluteError(i); } return ret / (double) numColumns(); }
/** * Average RSE across all columns * @return */ public double averagerelativeSquaredError() { double ret = 0.0; for (int i = 0; i < numColumns(); i++) { ret += relativeSquaredError(i); } return ret / (double) numColumns(); }
/** * Average RMSE across all columns * @return */ public double averagerootMeanSquaredError() { double ret = 0.0; for (int i = 0; i < numColumns(); i++) { ret += rootMeanSquaredError(i); } return ret / (double) numColumns(); }
/** * Average R2 across all columns * @return */ public double averagecorrelationR2() { double ret = 0.0; for (int i = 0; i < numColumns(); i++) { ret += correlationR2(i); } return ret / (double) numColumns(); } }
columnNames = createDefaultColumnNames(numColumns()); int maxLabelLength = 0; for (String s : columnNames)
public static double getScoreFromRegressionEval(RegressionEvaluation eval, RegressionValue regressionValue) { double sum = 0.0; int nColumns = eval.numColumns(); switch (regressionValue) { case MSE: for (int i = 0; i < nColumns; i++) sum += eval.meanSquaredError(i); break; case MAE: for (int i = 0; i < nColumns; i++) sum += eval.meanAbsoluteError(i); break; case RMSE: for (int i = 0; i < nColumns; i++) sum += eval.rootMeanSquaredError(i); break; case RSE: for (int i = 0; i < nColumns; i++) sum += eval.relativeSquaredError(i); break; case CorrCoeff: for (int i = 0; i < nColumns; i++) sum += eval.correlationR2(i); sum /= nColumns; break; } return sum; }
int totalColumns = 0; for (int i = 0; i < evaluations.length; i++) { int nColumns = evaluations[i].numColumns(); totalColumns += nColumns; sum += getScoreFromRegressionEval(evaluations[i], regressionValue);