public class LossFunctionGradientCheck
extends java.lang.Object
Constructor and Description |
---|
LossFunctionGradientCheck() |
Modifier and Type | Method and Description |
---|---|
static org.nd4j.linalg.api.ndarray.INDArray[] |
getFeaturesAndLabels(org.nd4j.linalg.lossfunctions.ILossFunction l,
int[] featuresShape,
int[] labelsShape,
long seed) |
static org.nd4j.linalg.api.ndarray.INDArray[] |
getFeaturesAndLabels(org.nd4j.linalg.lossfunctions.ILossFunction l,
int minibatch,
int nIn,
int nOut,
long seed) |
void |
lossFunctionGradientCheck() |
void |
lossFunctionGradientCheckLossLayer() |
void |
lossFunctionWeightedGradientCheck() |
public void lossFunctionGradientCheck()
public void lossFunctionGradientCheckLossLayer()
public static org.nd4j.linalg.api.ndarray.INDArray[] getFeaturesAndLabels(org.nd4j.linalg.lossfunctions.ILossFunction l, int minibatch, int nIn, int nOut, long seed)
public static org.nd4j.linalg.api.ndarray.INDArray[] getFeaturesAndLabels(org.nd4j.linalg.lossfunctions.ILossFunction l, int[] featuresShape, int[] labelsShape, long seed)
public void lossFunctionWeightedGradientCheck()