public class VAEReconErrorScoreCalculator extends BaseScoreCalculator<Model>
for reconstruction probability
,
Serialized FormModifier and Type | Field and Description |
---|---|
protected RegressionEvaluation |
evaluation |
protected RegressionEvaluation.Metric |
metric |
exampleCount, iterator, mdsIterator, minibatchCount, scoreSum
Constructor and Description |
---|
VAEReconErrorScoreCalculator(RegressionEvaluation.Metric metric,
DataSetIterator iterator)
Constructor for reconstruction *ERROR*
|
Modifier and Type | Method and Description |
---|---|
protected double |
finalScore(double scoreSum,
int minibatchCount,
int exampleCount) |
boolean |
minimizeScore() |
protected INDArray[] |
output(Model network,
INDArray[] input,
INDArray[] fMask,
INDArray[] lMask) |
protected INDArray |
output(Model net,
INDArray input,
INDArray fMask,
INDArray lMask) |
protected void |
reset() |
protected double |
scoreMinibatch(Model network,
INDArray[] features,
INDArray[] labels,
INDArray[] fMask,
INDArray[] lMask,
INDArray[] output) |
protected double |
scoreMinibatch(Model network,
INDArray features,
INDArray labels,
INDArray fMask,
INDArray lMask,
INDArray output) |
arr, calculateScore, get0
protected final RegressionEvaluation.Metric metric
protected RegressionEvaluation evaluation
public VAEReconErrorScoreCalculator(RegressionEvaluation.Metric metric, DataSetIterator iterator)
metric
- iterator
- protected void reset()
reset
in class BaseScoreCalculator<Model>
protected INDArray output(Model net, INDArray input, INDArray fMask, INDArray lMask)
output
in class BaseScoreCalculator<Model>
protected INDArray[] output(Model network, INDArray[] input, INDArray[] fMask, INDArray[] lMask)
output
in class BaseScoreCalculator<Model>
protected double scoreMinibatch(Model network, INDArray features, INDArray labels, INDArray fMask, INDArray lMask, INDArray output)
scoreMinibatch
in class BaseScoreCalculator<Model>
protected double scoreMinibatch(Model network, INDArray[] features, INDArray[] labels, INDArray[] fMask, INDArray[] lMask, INDArray[] output)
scoreMinibatch
in class BaseScoreCalculator<Model>
protected double finalScore(double scoreSum, int minibatchCount, int exampleCount)
finalScore
in class BaseScoreCalculator<Model>
public boolean minimizeScore()
Copyright © 2020. All rights reserved.