public class VAEReconErrorScoreCalculator extends BaseScoreCalculator<Model>
for reconstruction probability,
Serialized Form| Modifier 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, get0protected 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()