org.deeplearning4j.nn.modelimport.keras

Class KerasModelEndToEndTest

java.lang.Object

org.deeplearning4j.nn.modelimport.keras.KerasModelEndToEndTest

public class KerasModelEndToEndTest
extends java.lang.Object

Unit tests for end-to-end Keras model import. TODO: find a more elegant fix for the per-layer activation comparisons since some layers (e.g., ActivationLayer) overwrite previous layer's activations when run in train=false mode. TODO: ndarray.equalsWithEps(ndarray) appears to be broken

Field Summary

Fields

Modifier and Type	Field and Description
static double	EPS
static java.lang.String	GROUP_ACTIVATIONS
static java.lang.String	GROUP_ATTR_INPUTS
static java.lang.String	GROUP_ATTR_OUTPUTS
static java.lang.String	GROUP_PREDICTIONS
static java.lang.String	H5_EXTENSION
static java.lang.String	TEMP_MODEL_FILENAME
static java.lang.String	TEMP_OUTPUTS_FILENAME

Constructor Summary

Constructors

Constructor and Description
KerasModelEndToEndTest()

Method Summary

Modifier and Type	Method and Description
static void	compareINDArrays(java.lang.String label, org.nd4j.linalg.api.ndarray.INDArray a, org.nd4j.linalg.api.ndarray.INDArray b, double eps)
static void	compareMulticlassAUC(java.lang.String label, org.nd4j.linalg.api.ndarray.INDArray target, org.nd4j.linalg.api.ndarray.INDArray a, org.nd4j.linalg.api.ndarray.INDArray b, int nbClasses, double eps)
static java.util.Map<java.lang.String,org.nd4j.linalg.api.ndarray.INDArray>	getActivations(Hdf5Archive archive, boolean tensorFlowImageDimOrdering)
static org.nd4j.linalg.api.ndarray.INDArray[]	getInputs(Hdf5Archive archive, boolean tensorFlowImageDimOrdering)
static org.nd4j.linalg.api.ndarray.INDArray[]	getOutputs(Hdf5Archive archive, boolean tensorFlowImageDimOrdering)
static org.nd4j.linalg.api.ndarray.INDArray[]	getPredictions(Hdf5Archive archive, boolean tensorFlowImageDimOrdering)
void	importMnistMlpTensorFlowEndToEndModelTest()

Methods inherited from class java.lang.Object

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

Field Detail

GROUP_ATTR_INPUTS

public static final java.lang.String GROUP_ATTR_INPUTS

See Also:

Constant Field Values

GROUP_ATTR_OUTPUTS

public static final java.lang.String GROUP_ATTR_OUTPUTS

See Also:

Constant Field Values

GROUP_PREDICTIONS

public static final java.lang.String GROUP_PREDICTIONS

See Also:

Constant Field Values

GROUP_ACTIVATIONS

public static final java.lang.String GROUP_ACTIVATIONS

See Also:

Constant Field Values

TEMP_OUTPUTS_FILENAME

public static final java.lang.String TEMP_OUTPUTS_FILENAME

See Also:

Constant Field Values

TEMP_MODEL_FILENAME

public static final java.lang.String TEMP_MODEL_FILENAME

See Also:

Constant Field Values

H5_EXTENSION

public static final java.lang.String H5_EXTENSION

See Also:

Constant Field Values

EPS

public static final double EPS

See Also:

Constant Field Values

Constructor Detail

KerasModelEndToEndTest

public KerasModelEndToEndTest()

Method Detail

importMnistMlpTensorFlowEndToEndModelTest

public void importMnistMlpTensorFlowEndToEndModelTest()
                                               throws java.lang.Exception

Throws:

java.lang.Exception

getInputs

public static org.nd4j.linalg.api.ndarray.INDArray[] getInputs(Hdf5Archive archive,
                                                               boolean tensorFlowImageDimOrdering)
                                                        throws java.lang.Exception

Throws:

java.lang.Exception

getActivations

public static java.util.Map<java.lang.String,org.nd4j.linalg.api.ndarray.INDArray> getActivations(Hdf5Archive archive,
                                                                                                  boolean tensorFlowImageDimOrdering)
                                                                                           throws java.lang.Exception

Throws:

java.lang.Exception

getOutputs

public static org.nd4j.linalg.api.ndarray.INDArray[] getOutputs(Hdf5Archive archive,
                                                                boolean tensorFlowImageDimOrdering)
                                                         throws java.lang.Exception

Throws:

java.lang.Exception

getPredictions

public static org.nd4j.linalg.api.ndarray.INDArray[] getPredictions(Hdf5Archive archive,
                                                                    boolean tensorFlowImageDimOrdering)
                                                             throws java.lang.Exception

Throws:

java.lang.Exception

compareINDArrays

public static void compareINDArrays(java.lang.String label,
                                    org.nd4j.linalg.api.ndarray.INDArray a,
                                    org.nd4j.linalg.api.ndarray.INDArray b,
                                    double eps)

compareMulticlassAUC

public static void compareMulticlassAUC(java.lang.String label,
                                        org.nd4j.linalg.api.ndarray.INDArray target,
                                        org.nd4j.linalg.api.ndarray.INDArray a,
                                        org.nd4j.linalg.api.ndarray.INDArray b,
                                        int nbClasses,
                                        double eps)