org.deeplearning4j.zoo

Interface InstantiableModel

All Known Implementing Classes:

AlexNet, FaceNetNN4Small2, GoogLeNet, InceptionResNetV1, LeNet, ResNet50, SimpleCNN, TextGenerationLSTM, VGG16, VGG19, ZooModel

public interface InstantiableModel

Interface for defining a model that can be instantiated and return information about itself.

Method Summary

Modifier and Type	Method and Description
Model	init()
ModelMetaData	metaData()
Class<? extends Model>	modelType()
long	pretrainedChecksum(PretrainedType pretrainedType)
String	pretrainedUrl(PretrainedType pretrainedType)
void	setInputShape(int[][] inputShape)
ZooType	zooType()

Method Detail

setInputShape

void setInputShape(int[][] inputShape)

init

Model init()

metaData

ModelMetaData metaData()

zooType

ZooType zooType()

modelType

Class<? extends Model> modelType()

pretrainedUrl

String pretrainedUrl(PretrainedType pretrainedType)

pretrainedChecksum

long pretrainedChecksum(PretrainedType pretrainedType)