public enum PretrainedType extends java.lang.Enum<PretrainedType> implements ApiWrapper<org.deeplearning4j.zoo.PretrainedType>
PretrainedType
. This is necessary as Weka's
run script cannot find the enum classes during the option parsing as they reside in the Dl4j
backend and are at that time not visible to the class loader.Enum Constant and Description |
---|
IMAGENET
Pretrained on ImageNet
|
MNIST
Pretrained on MNIST.
|
NONE
Not pretrained.
|
VGGFACE
Pretrained on VGGFace.
|
Modifier and Type | Method and Description |
---|---|
static PretrainedType |
fromBackend(org.deeplearning4j.zoo.PretrainedType pretrainedType)
Parse backend pooling type and return weka enum implementation.
|
org.deeplearning4j.zoo.PretrainedType |
getBackend()
Access the DL4J backend.
|
void |
initializeBackend()
Initialize the DL4J backend.
|
void |
setBackend(org.deeplearning4j.zoo.PretrainedType pretrainedType)
Set the DL4J backend.
|
static PretrainedType |
valueOf(java.lang.String name)
Returns the enum constant of this type with the specified name.
|
static PretrainedType[] |
values()
Returns an array containing the constants of this enum type, in
the order they are declared.
|
public static final PretrainedType IMAGENET
public static final PretrainedType MNIST
public static final PretrainedType VGGFACE
public static final PretrainedType NONE
public static PretrainedType[] values()
for (PretrainedType c : PretrainedType.values()) System.out.println(c);
public static PretrainedType valueOf(java.lang.String name)
name
- the name of the enum constant to be returned.java.lang.IllegalArgumentException
- if this enum type has no constant with the specified namejava.lang.NullPointerException
- if the argument is nullpublic static PretrainedType fromBackend(org.deeplearning4j.zoo.PretrainedType pretrainedType)
pretrainedType
- Pooling typepublic org.deeplearning4j.zoo.PretrainedType getBackend()
ApiWrapper
getBackend
in interface ApiWrapper<org.deeplearning4j.zoo.PretrainedType>
public void setBackend(org.deeplearning4j.zoo.PretrainedType pretrainedType)
ApiWrapper
setBackend
in interface ApiWrapper<org.deeplearning4j.zoo.PretrainedType>
pretrainedType
- Backend that should be wrapped by this classpublic void initializeBackend()
ApiWrapper
initializeBackend
in interface ApiWrapper<org.deeplearning4j.zoo.PretrainedType>