org.deeplearning4j.eval

Class RegressionEvaluation

java.lang.Object

org.deeplearning4j.eval.BaseEvaluation<RegressionEvaluation>

org.deeplearning4j.eval.RegressionEvaluation

All Implemented Interfaces:

java.io.Serializable, IEvaluation<RegressionEvaluation>

public class RegressionEvaluation
extends BaseEvaluation<RegressionEvaluation>

Evaluation method for the evaluation of regression algorithms.
Provides the following metrics, for each column:
- MSE: mean squared error
- MAE: mean absolute error
- RMSE: root mean squared error
- RSE: relative squared error
- correlation coefficient
See for example: http://www.saedsayad.com/model_evaluation_r.htm For classification, see Evaluation

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
static int	DEFAULT_PRECISION

Constructor Summary

Constructors

Constructor and Description
RegressionEvaluation(int nColumns) Create a regression evaluation object with the specified number of columns, and default precision for the stats() method.
RegressionEvaluation(int nColumns, int precision) Create a regression evaluation object with the specified number of columns, and specified precision for the stats() method.
RegressionEvaluation(java.util.List<java.lang.String> columnNames) Create a regression evaluation object with default precision for the stats() method
RegressionEvaluation(java.util.List<java.lang.String> columnNames, int precision) Create a regression evaluation object with specified precision for the stats() method
RegressionEvaluation(java.lang.String... columnNames) Create a regression evaluation object with default precision for the stats() method

Method Summary

Modifier and Type	Method and Description
double	correlationR2(int column)
void	eval(org.nd4j.linalg.api.ndarray.INDArray labels, org.nd4j.linalg.api.ndarray.INDArray predictions)
double	meanAbsoluteError(int column)
double	meanSquaredError(int column)
void	merge(RegressionEvaluation other)
int	numColumns()
double	relativeSquaredError(int column)
double	rootMeanSquaredError(int column)
java.lang.String	stats()

Methods inherited from class org.deeplearning4j.eval.BaseEvaluation

eval, eval, evalTimeSeries, evalTimeSeries

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

DEFAULT_PRECISION

public static final int DEFAULT_PRECISION

See Also:

Constant Field Values

Constructor Detail

RegressionEvaluation

public RegressionEvaluation(int nColumns)

Create a regression evaluation object with the specified number of columns, and default precision for the stats() method.

Parameters:

nColumns - Number of columns

RegressionEvaluation

public RegressionEvaluation(int nColumns,
                            int precision)

Create a regression evaluation object with the specified number of columns, and specified precision for the stats() method.

Parameters:

nColumns - Number of columns

RegressionEvaluation

public RegressionEvaluation(java.lang.String... columnNames)

Create a regression evaluation object with default precision for the stats() method

Parameters:

columnNames - Names of the columns

RegressionEvaluation

public RegressionEvaluation(java.util.List<java.lang.String> columnNames)

Create a regression evaluation object with default precision for the stats() method

Parameters:

columnNames - Names of the columns

RegressionEvaluation

public RegressionEvaluation(java.util.List<java.lang.String> columnNames,
                            int precision)

Create a regression evaluation object with specified precision for the stats() method

Parameters:

columnNames - Names of the columns

Method Detail

eval

public void eval(org.nd4j.linalg.api.ndarray.INDArray labels,
                 org.nd4j.linalg.api.ndarray.INDArray predictions)

merge

public void merge(RegressionEvaluation other)

stats

public java.lang.String stats()

numColumns

public int numColumns()

meanSquaredError

public double meanSquaredError(int column)

meanAbsoluteError

public double meanAbsoluteError(int column)

rootMeanSquaredError

public double rootMeanSquaredError(int column)

correlationR2

public double correlationR2(int column)

relativeSquaredError

public double relativeSquaredError(int column)