org.deeplearning4j.arbiter.scoring.impl

Class EvaluationScoreFunction

java.lang.Object

org.deeplearning4j.arbiter.scoring.impl.BaseNetScoreFunction

org.deeplearning4j.arbiter.scoring.impl.EvaluationScoreFunction

All Implemented Interfaces:

Serializable, ScoreFunction

public class EvaluationScoreFunction
extends BaseNetScoreFunction

Score function that calculates an evaluation Evaluation.Metric on the test set for a MultiLayerNetwork or ComputationGraph

Author:

Alex Black

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected org.nd4j.evaluation.classification.Evaluation.Metric	metric

Constructor Summary

Constructors

Constructor and Description
EvaluationScoreFunction(org.deeplearning4j.eval.Evaluation.Metric metric)
EvaluationScoreFunction(org.nd4j.evaluation.classification.Evaluation.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.classification.Evaluation.Metric metric

Constructor Detail

EvaluationScoreFunction

public EvaluationScoreFunction(@NonNull
                               org.deeplearning4j.eval.Evaluation.Metric metric)

Parameters:

metric - Evaluation metric to calculate

EvaluationScoreFunction

public EvaluationScoreFunction(@NonNull
                               org.nd4j.evaluation.classification.Evaluation.Metric metric)

Parameters:

metric - Evaluation metric to calculate

Method Detail

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

minimize

public boolean minimize()