org.deeplearning4j.nn.api

Interface NeuralNetworkPrototype

public interface NeuralNetworkPrototype

Author:

Alex Black, raver119@gmail.com

Method Summary

Modifier and Type	Method and Description
Map<String,org.nd4j.linalg.api.ndarray.INDArray>	activations(org.nd4j.linalg.api.ndarray.INDArray input)
void	fit(org.nd4j.linalg.dataset.api.DataSet dataSet)
void	fit(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)
<T> T	getConfiguration()
Layer[]	getLayers()
org.nd4j.linalg.api.ndarray.INDArray	getParams()
double	getScore()
Updater	getUpdater()
org.nd4j.linalg.api.ndarray.INDArray[]	output(org.nd4j.linalg.api.ndarray.INDArray... input)
org.nd4j.linalg.api.ndarray.INDArray	output(org.nd4j.linalg.api.ndarray.INDArray input)
void	pretrain(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)
void	rnnClearPreviousState()
Map<String,Map<String,org.nd4j.linalg.api.ndarray.INDArray>>	rnnGetPreviousStates()
void	rnnTimeStep(org.nd4j.linalg.api.ndarray.INDArray... input)

Method Detail

getParams

org.nd4j.linalg.api.ndarray.INDArray getParams()

getUpdater

Updater getUpdater()

getScore

double getScore()

getConfiguration

<T> T getConfiguration()

getLayers

Layer[] getLayers()

fit

void fit(org.nd4j.linalg.dataset.api.DataSet dataSet)

fit

void fit(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)

pretrain

void pretrain(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)

activations

Map<String,org.nd4j.linalg.api.ndarray.INDArray> activations(org.nd4j.linalg.api.ndarray.INDArray input)

output

org.nd4j.linalg.api.ndarray.INDArray output(org.nd4j.linalg.api.ndarray.INDArray input)

output

org.nd4j.linalg.api.ndarray.INDArray[] output(org.nd4j.linalg.api.ndarray.INDArray... input)

rnnClearPreviousState

void rnnClearPreviousState()

rnnGetPreviousStates

Map<String,Map<String,org.nd4j.linalg.api.ndarray.INDArray>> rnnGetPreviousStates()

rnnTimeStep

void rnnTimeStep(org.nd4j.linalg.api.ndarray.INDArray... input)