org.deeplearning4j.zoo.model

Class TextGenerationLSTM

java.lang.Object

org.deeplearning4j.zoo.ZooModel

org.deeplearning4j.zoo.model.TextGenerationLSTM

All Implemented Interfaces:

InstantiableModel

public class TextGenerationLSTM
extends ZooModel

LSTM designed for text generation. Can be trained on a corpus of text. For this model, numClasses is used to input totalUniqueCharacters for the LSTM input layer. Architecture follows this implementation: https://github.com/fchollet/keras/blob/master/examples/lstm_text_generation.py

Walt Whitman weights are available for generating text from his works, adapted from https://github.com/craigomac/InfiniteMonkeys.

Author:

Justin Long (crockpotveggies)

Method Summary

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

Methods inherited from class org.deeplearning4j.zoo.ZooModel

initPretrained, initPretrained, modelName, pretrainedAvailable

Methods inherited from class java.lang.Object

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

Method Detail

pretrainedUrl

public String pretrainedUrl(PretrainedType pretrainedType)

pretrainedChecksum

public long pretrainedChecksum(PretrainedType pretrainedType)

modelType

public Class<? extends Model> modelType()

conf

public MultiLayerConfiguration conf()

init

public Model init()

metaData

public ModelMetaData metaData()

setInputShape

public void setInputShape(int[][] inputShape)