public class TestCompGraphCNN
extends java.lang.Object
Modifier and Type | Field and Description |
---|---|
protected ComputationGraphConfiguration |
conf |
protected org.nd4j.linalg.dataset.api.iterator.DataSetIterator |
dataSetIterator |
protected org.nd4j.linalg.dataset.DataSet |
ds |
protected ComputationGraph |
graph |
Constructor and Description |
---|
TestCompGraphCNN() |
Modifier and Type | Method and Description |
---|---|
void |
beforeDo() |
protected static org.nd4j.linalg.dataset.api.iterator.DataSetIterator |
getDS() |
protected static ComputationGraphConfiguration |
getMultiInputGraphConfig() |
protected static int |
getNumParams() |
void |
testBackwardIrisBasic() |
void |
testCNNComputationGraphKernelTooLarge() |
void |
testCNNComputationGraphSingleOutFeatureMap() |
void |
testCnnLRN_BN() |
void |
testConfigBasic() |
void |
testForwardBasic() |
protected ComputationGraphConfiguration conf
protected ComputationGraph graph
protected org.nd4j.linalg.dataset.api.iterator.DataSetIterator dataSetIterator
protected org.nd4j.linalg.dataset.DataSet ds
protected static ComputationGraphConfiguration getMultiInputGraphConfig()
protected static org.nd4j.linalg.dataset.api.iterator.DataSetIterator getDS()
protected static int getNumParams()
public void beforeDo()
public void testConfigBasic()
public void testForwardBasic()
public void testBackwardIrisBasic()
public void testCNNComputationGraphKernelTooLarge()
public void testCNNComputationGraphSingleOutFeatureMap()
public void testCnnLRN_BN()