ai.libs.jaicore.ml.dyadranking.zeroshot.util

Class InputOptListener

java.lang.Object

ai.libs.jaicore.ml.dyadranking.zeroshot.util.InputOptListener

public class InputOptListener
extends java.lang.Object

Constructor Summary

Constructors

Constructor and Description
InputOptListener(int[] indicesToWatch)

Method Summary

Modifier and Type	Method and Description
java.util.List<org.nd4j.linalg.api.ndarray.INDArray>	getInputList()
java.util.List<java.lang.Double>	getOutputList()
void	reportOptimizationStep(org.nd4j.linalg.api.ndarray.INDArray plNetInput, double plNetOutput)

Methods inherited from class java.lang.Object

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

Constructor Detail

InputOptListener

public InputOptListener(int[] indicesToWatch)

Method Detail

reportOptimizationStep

public void reportOptimizationStep(org.nd4j.linalg.api.ndarray.INDArray plNetInput,
                                   double plNetOutput)

getInputList

public java.util.List<org.nd4j.linalg.api.ndarray.INDArray> getInputList()

getOutputList

public java.util.List<java.lang.Double> getOutputList()