public abstract class BaseScoreCalculator<T extends Model> extends Object implements ScoreCalculator<T>
Modifier and Type | Field and Description |
---|---|
protected int |
exampleCount |
protected DataSetIterator |
iterator |
protected MultiDataSetIterator |
mdsIterator |
protected int |
minibatchCount |
protected double |
scoreSum |
Modifier | Constructor and Description |
---|---|
protected |
BaseScoreCalculator(DataSetIterator iterator) |
protected |
BaseScoreCalculator(MultiDataSetIterator iterator) |
Modifier and Type | Method and Description |
---|---|
static INDArray[] |
arr(INDArray in) |
double |
calculateScore(T network)
Calculate the score for the given MultiLayerNetwork
|
protected abstract double |
finalScore(double scoreSum,
int minibatchCount,
int exampleCount) |
static INDArray |
get0(INDArray[] in) |
protected abstract INDArray[] |
output(T network,
INDArray[] input,
INDArray[] fMask,
INDArray[] lMask) |
protected abstract INDArray |
output(T network,
INDArray input,
INDArray fMask,
INDArray lMask) |
protected abstract void |
reset() |
protected abstract double |
scoreMinibatch(T network,
INDArray[] features,
INDArray[] labels,
INDArray[] fMask,
INDArray[] lMask,
INDArray[] output) |
protected double |
scoreMinibatch(T network,
INDArray features,
INDArray labels,
INDArray fMask,
INDArray lMask,
INDArray output) |
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
minimizeScore
protected MultiDataSetIterator mdsIterator
protected DataSetIterator iterator
protected double scoreSum
protected int minibatchCount
protected int exampleCount
protected BaseScoreCalculator(@NonNull DataSetIterator iterator)
protected BaseScoreCalculator(@NonNull MultiDataSetIterator iterator)
public double calculateScore(T network)
ScoreCalculator
calculateScore
in interface ScoreCalculator<T extends Model>
protected abstract void reset()
protected abstract INDArray output(T network, INDArray input, INDArray fMask, INDArray lMask)
protected abstract INDArray[] output(T network, INDArray[] input, INDArray[] fMask, INDArray[] lMask)
protected double scoreMinibatch(T network, INDArray features, INDArray labels, INDArray fMask, INDArray lMask, INDArray output)
protected abstract double scoreMinibatch(T network, INDArray[] features, INDArray[] labels, INDArray[] fMask, INDArray[] lMask, INDArray[] output)
protected abstract double finalScore(double scoreSum, int minibatchCount, int exampleCount)
Copyright © 2018. All rights reserved.