RegressionEvaluation (deeplearning4j-nn 1.0.0-beta API)

java.lang.Object
- org.deeplearning4j.eval.BaseEvaluation<RegressionEvaluation>
- - org.deeplearning4j.eval.RegressionEvaluation

All Implemented Interfaces:

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
- PC: pearson correlation coefficient
- R^2: coefficient of determination See for example: http://www.saedsayad.com/model_evaluation_r.htm For classification, see Evaluation

Author:

Alex Black

See Also:

Serialized Form

Nested Class Summary

Nested Classes
Modifier and Type Class and Description

static class RegressionEvaluation.Metric

Nested Classes
Modifier and Type	Class and Description
`static class`	`RegressionEvaluation.Metric`

Field Summary

Fields
Modifier and Type Field and Description

static int DEFAULT_PRECISION

Fields
Modifier and Type	Field and Description
`static int`	`DEFAULT_PRECISION`

Constructor Summary

Constructors
Constructor and Description
`RegressionEvaluation()`
`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(List<String> columnNames)` Create a regression evaluation object with default precision for the stats() method
`RegressionEvaluation(List<String> columnNames, int precision)` Create a regression evaluation object with specified precision for the stats() method
`RegressionEvaluation(String... columnNames)` Create a regression evaluation object with default precision for the stats() method

Method Summary

All Methods Instance Methods Concrete Methods Deprecated Methods
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(org.nd4j.linalg.api.ndarray.INDArray labels, org.nd4j.linalg.api.ndarray.INDArray predictions)`
`void`	`eval(org.nd4j.linalg.api.ndarray.INDArray labels, org.nd4j.linalg.api.ndarray.INDArray predictions, org.nd4j.linalg.api.ndarray.INDArray maskArray)`
`double`	`meanAbsoluteError(int column)`
`double`	`meanSquaredError(int column)`
`void`	`merge(RegressionEvaluation other)`
`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)`
`String`	`stats()`

Methods inherited from class org.deeplearning4j.eval.BaseEvaluation
eval, evalTimeSeries, evalTimeSeries, fromJson, fromYaml, toJson, toString, toYaml

Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

- Field Detail
  - DEFAULT_PRECISION
```
public static final int DEFAULT_PRECISION
```
    See Also:
    
    Constant Field Values
- Constructor Detail
  - RegressionEvaluation
```
public RegressionEvaluation()
```
  - 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(String... columnNames)
```
    Create a regression evaluation object with default precision for the stats() method
    
    Parameters:
    
    columnNames - Names of the columns
  - RegressionEvaluation
```
public RegressionEvaluation(List<String> columnNames)
```
    Create a regression evaluation object with default precision for the stats() method
    
    Parameters:
    
    columnNames - Names of the columns
  - RegressionEvaluation
```
public RegressionEvaluation(List<String> columnNames,
                            int precision)
```
    Create a regression evaluation object with specified precision for the stats() method
    
    Parameters:
    
    columnNames - Names of the columns
- Method Detail
  - reset
```
public void reset()
```
  - eval
```
public void eval(org.nd4j.linalg.api.ndarray.INDArray labels,
                 org.nd4j.linalg.api.ndarray.INDArray predictions)
```
  - eval
```
public void eval(org.nd4j.linalg.api.ndarray.INDArray labels,
                 org.nd4j.linalg.api.ndarray.INDArray predictions,
                 org.nd4j.linalg.api.ndarray.INDArray maskArray)
```
    Specified by:
    
    eval in interface IEvaluation<RegressionEvaluation>
    
    Overrides:
    
    eval in class BaseEvaluation<RegressionEvaluation>
  - merge
```
public void merge(RegressionEvaluation other)
```
  - stats
```
public String stats()
```
    Returns:
  - numColumns
```
public int numColumns()
```
  - meanSquaredError
```
public double meanSquaredError(int column)
```
  - meanAbsoluteError
```
public double meanAbsoluteError(int column)
```
  - rootMeanSquaredError
```
public double rootMeanSquaredError(int column)
```
  - correlationR2
```
@Deprecated
public double correlationR2(int column)
```
    Deprecated. Use pearsonCorrelation(int) instead. For the R2 score use rSquared(int).
    
    Legacy method for the correlation score.
    
    Parameters:
    
    column - Column to evaluate
    
    Returns:
    
    Pearson Correlation for the given column
    
    See Also:
    
    #pearsonCorrelation(int)}
  - pearsonCorrelation
```
public double pearsonCorrelation(int column)
```
    Pearson Correlation Coefficient for samples
    
    Parameters:
    
    column - Column to evaluate
    
    Returns:
    
    Pearson Correlation Coefficient for column with index column
    
    See Also:
    
    Wikipedia
  - rSquared
```
public double rSquared(int column)
```
    Coefficient of Determination (R^2 Score)
    
    Parameters:
    
    column - Column to evaluate
    
    Returns:
    
    R^2 score for column with index column
    
    See Also:
    
    Wikipedia
  - relativeSquaredError
```
public double relativeSquaredError(int column)
```
  - averageMeanSquaredError
```
public double averageMeanSquaredError()
```
    Average MSE across all columns
    
    Returns:
  - averageMeanAbsoluteError
```
public double averageMeanAbsoluteError()
```
    Average MAE across all columns
    
    Returns:
  - averagerootMeanSquaredError
```
public double averagerootMeanSquaredError()
```
    Average RMSE across all columns
    
    Returns:
  - averagerelativeSquaredError
```
public double averagerelativeSquaredError()
```
    Average RSE across all columns
    
    Returns:
  - averagecorrelationR2
```
@Deprecated
public double averagecorrelationR2()
```
    Deprecated. Use averagePearsonCorrelation() instead. For the R2 score use averageRSquared().
    
    Legacy method for the correlation average across all columns.
    
    Returns:
    
    Pearson Correlation averaged over all columns
    
    See Also:
    
    #averagePearsonCorrelation()}
  - averagePearsonCorrelation
```
public double averagePearsonCorrelation()
```
    Average Pearson Correlation Coefficient across all columns
    
    Returns:
    
    Pearson Correlation Coefficient across all columns
  - averageRSquared
```
public double averageRSquared()
```
    Average R2 across all columns
    
    Returns:
    
    R2 score accross all columns
  - scoreForMetric
```
public double scoreForMetric(RegressionEvaluation.Metric metric)
```

Class RegressionEvaluation

Nested Class Summary

Field Summary

Constructor Summary

Method Summary

Methods inherited from class org.deeplearning4j.eval.BaseEvaluation

Methods inherited from class java.lang.Object

Field Detail

DEFAULT_PRECISION

Constructor Detail

RegressionEvaluation

RegressionEvaluation

RegressionEvaluation

RegressionEvaluation

RegressionEvaluation

RegressionEvaluation

Method Detail

reset

eval

eval

merge

stats

numColumns

meanSquaredError

meanAbsoluteError

rootMeanSquaredError

correlationR2

pearsonCorrelation

rSquared

relativeSquaredError

averageMeanSquaredError

averageMeanAbsoluteError

averagerootMeanSquaredError

averagerelativeSquaredError

averagecorrelationR2

averagePearsonCorrelation

averageRSquared

scoreForMetric