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()