org.deeplearning4j.arbiter.scoring.impl

Class RegressionScoreFunction

java.lang.Object

org.deeplearning4j.arbiter.scoring.impl.BaseNetScoreFunction

org.deeplearning4j.arbiter.scoring.impl.RegressionScoreFunction

All Implemented Interfaces:

Serializable, ScoreFunction

public class RegressionScoreFunction
extends BaseNetScoreFunction

Score function for regression (including multi-label regression) for a MultiLayerNetwork or ComputationGraph on a test set. Supports all regression metrics: RegressionEvaluation.Metric

Author:

Alex Black

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected org.nd4j.evaluation.regression.RegressionEvaluation.Metric	metric

Constructor Summary

Constructors

Constructor and Description
RegressionScoreFunction(org.deeplearning4j.eval.RegressionEvaluation.Metric metric)
RegressionScoreFunction(@NonNull org.nd4j.evaluation.regression.RegressionEvaluation.Metric metric)

Method Summary

Modifier and Type	Method and Description
boolean	minimize()
double	score(org.deeplearning4j.nn.graph.ComputationGraph graph, org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)
double	score(org.deeplearning4j.nn.graph.ComputationGraph graph, org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator)
double	score(org.deeplearning4j.nn.multilayer.MultiLayerNetwork net, org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)
double	score(org.deeplearning4j.nn.multilayer.MultiLayerNetwork net, org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator)
String	toString()

Methods inherited from class org.deeplearning4j.arbiter.scoring.impl.BaseNetScoreFunction

getSupportedDataTypes, getSupportedModelTypes, score, score, score

Methods inherited from class java.lang.Object

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

Field Detail

metric

protected org.nd4j.evaluation.regression.RegressionEvaluation.Metric metric

Constructor Detail

RegressionScoreFunction

public RegressionScoreFunction(@NonNull
                               org.deeplearning4j.eval.RegressionEvaluation.Metric metric)

RegressionScoreFunction

public RegressionScoreFunction(@NonNull
                               @NonNull org.nd4j.evaluation.regression.RegressionEvaluation.Metric metric)

Method Detail

minimize

public boolean minimize()

toString

public String toString()

Overrides:

toString in class Object

score

public double score(org.deeplearning4j.nn.multilayer.MultiLayerNetwork net,
                    org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)

Specified by:

score in class BaseNetScoreFunction

score

public double score(org.deeplearning4j.nn.multilayer.MultiLayerNetwork net,
                    org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator)

Specified by:

score in class BaseNetScoreFunction

score

public double score(org.deeplearning4j.nn.graph.ComputationGraph graph,
                    org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)

Specified by:

score in class BaseNetScoreFunction

score

public double score(org.deeplearning4j.nn.graph.ComputationGraph graph,
                    org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator)

Specified by:

score in class BaseNetScoreFunction