public class RegressionEvaluation extends BaseEvaluation<RegressionEvaluation>
| Modifier and Type | Class and Description |
|---|---|
static class |
RegressionEvaluation.Metric |
| Modifier and Type | Field and Description |
|---|---|
protected int |
axis |
static int |
DEFAULT_PRECISION |
| Modifier | Constructor and Description |
|---|---|
|
RegressionEvaluation() |
protected |
RegressionEvaluation(int axis,
List<String> columnNames,
long precision) |
|
RegressionEvaluation(List<String> columnNames)
Create a regression evaluation object with default precision for the stats() method
|
|
RegressionEvaluation(List<String> columnNames,
long precision)
Create a regression evaluation object with specified precision for the stats() method
|
|
RegressionEvaluation(long nColumns)
Create a regression evaluation object with the specified number of columns, and default precision
for the stats() method.
|
|
RegressionEvaluation(long nColumns,
long precision)
Create a regression evaluation object with the specified number of columns, and specified precision
for the stats() method.
|
|
RegressionEvaluation(String... columnNames)
Create a regression evaluation object with default precision for the stats() method
|
| Modifier and Type | Method and Description |
|---|---|
double |
averagecorrelationR2()
Deprecated.
Use
averagePearsonCorrelation() instead.
For the R2 score use averageRSquared(). |
double |
averageMeanAbsoluteError()
Average MAE across all columns
|
double |
averageMeanSquaredError()
Average MSE across all columns
|
double |
averagePearsonCorrelation()
Average Pearson Correlation Coefficient across all columns
|
double |
averagerelativeSquaredError()
Average RSE across all columns
|
double |
averagerootMeanSquaredError()
Average RMSE across all columns
|
double |
averageRSquared()
Average R2 across all columns
|
double |
correlationR2(int column)
Deprecated.
Use
pearsonCorrelation(int) instead.
For the R2 score use rSquared(int). |
void |
eval(INDArray labels,
INDArray predictions) |
void |
eval(INDArray labelsArr,
INDArray predictionsArr,
INDArray maskArr) |
void |
eval(INDArray labels,
INDArray networkPredictions,
INDArray maskArray,
List<? extends Serializable> recordMetaData) |
static RegressionEvaluation |
fromJson(String json) |
int |
getAxis()
Get the axis - see
setAxis(int) for details |
double |
getValue(IMetric metric)
Get the value of a given metric for this evaluation.
|
double |
meanAbsoluteError(int column) |
double |
meanSquaredError(int column) |
void |
merge(RegressionEvaluation other) |
RegressionEvaluation |
newInstance()
Get a new instance of this evaluation, with the same configuration but no data.
|
int |
numColumns() |
double |
pearsonCorrelation(int column)
Pearson Correlation Coefficient for samples
|
double |
relativeSquaredError(int column) |
void |
reset() |
double |
rootMeanSquaredError(int column) |
double |
rSquared(int column)
Coefficient of Determination (R^2 Score)
|
double |
scoreForMetric(RegressionEvaluation.Metric metric) |
void |
setAxis(int axis)
Set the axis for evaluation - this is the dimension along which the probability (and label classes) are present.
For DL4J, this can be left as the default setting (axis = 1). Axis should be set as follows: For 2D (OutputLayer), shape [minibatch, numClasses] - axis = 1 For 3D, RNNs/CNN1D (DL4J RnnOutputLayer), NCW format, shape [minibatch, numClasses, sequenceLength] - axis = 1 For 3D, RNNs/CNN1D (DL4J RnnOutputLayer), NWC format, shape [minibatch, sequenceLength, numClasses] - axis = 2 For 4D, CNN2D (DL4J CnnLossLayer), NCHW format, shape [minibatch, channels, height, width] - axis = 1 For 4D, CNN2D, NHWC format, shape [minibatch, height, width, channels] - axis = 3 |
String |
stats() |
attempFromLegacyFromJson, eval, evalTimeSeries, evalTimeSeries, fromJson, fromYaml, reshapeAndExtractNotMasked, toJson, toString, toYamlpublic static final int DEFAULT_PRECISION
protected int axis
protected RegressionEvaluation(int axis,
List<String> columnNames,
long precision)
public RegressionEvaluation()
public RegressionEvaluation(long nColumns)
nColumns - Number of columnspublic RegressionEvaluation(long nColumns,
long precision)
nColumns - Number of columnspublic RegressionEvaluation(String... columnNames)
columnNames - Names of the columnspublic RegressionEvaluation(List<String> columnNames)
columnNames - Names of the columnspublic void setAxis(int axis)
axis - Axis to use for evaluationpublic int getAxis()
setAxis(int) for detailspublic void reset()
public void eval(INDArray labels, INDArray predictions)
eval in interface IEvaluation<RegressionEvaluation>eval in class BaseEvaluation<RegressionEvaluation>public void eval(INDArray labels, INDArray networkPredictions, INDArray maskArray, List<? extends Serializable> recordMetaData)
public void eval(INDArray labelsArr, INDArray predictionsArr, INDArray maskArr)
eval in interface IEvaluation<RegressionEvaluation>eval in class BaseEvaluation<RegressionEvaluation>public void merge(RegressionEvaluation other)
public String stats()
public int numColumns()
public double meanSquaredError(int column)
public double meanAbsoluteError(int column)
public double rootMeanSquaredError(int column)
@Deprecated public double correlationR2(int column)
pearsonCorrelation(int) instead.
For the R2 score use rSquared(int).column - Column to evaluate#pearsonCorrelation(int)}public double pearsonCorrelation(int column)
column - Column to evaluatecolumnpublic double rSquared(int column)
column - Column to evaluatecolumnpublic double relativeSquaredError(int column)
public double averageMeanSquaredError()
public double averageMeanAbsoluteError()
public double averagerootMeanSquaredError()
public double averagerelativeSquaredError()
@Deprecated public double averagecorrelationR2()
averagePearsonCorrelation() instead.
For the R2 score use averageRSquared().#averagePearsonCorrelation()}public double averagePearsonCorrelation()
public double averageRSquared()
public double getValue(IMetric metric)
IEvaluationpublic double scoreForMetric(RegressionEvaluation.Metric metric)
public static RegressionEvaluation fromJson(String json)
public RegressionEvaluation newInstance()
IEvaluationCopyright © 2021. All rights reserved.