Package org.deeplearning4j.nn.api
Interface NeuralNetwork
-
- All Known Implementing Classes:
ComputationGraph
,MultiLayerNetwork
public interface NeuralNetwork
- Author:
- raver119
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method Description <T extends IEvaluation>
T[]doEvaluation(DataSetIterator iterator, T... evaluations)
This method executes evaluation of the model against given iterator and evaluation implementations<T extends IEvaluation>
T[]doEvaluation(MultiDataSetIterator iterator, T... evaluations)
This method executes evaluation of the model against given iterator and evaluation implementationsvoid
fit(DataSet dataSet)
This method fits model with a given DataSetvoid
fit(DataSetIterator iterator)
This method fits model with a given DataSetIteratorvoid
fit(MultiDataSetIterator iterator)
This method fits model with a given MultiDataSetIteratorvoid
fit(MultiDataSet dataSet)
This method fits model with a given MultiDataSetConvexOptimizer
getOptimizer()
This method returns Optimizer used for trainingvoid
init()
This method does initialization of model PLEASE NOTE: All implementations should track own state, to avoid double spendingINDArray
params()
This method returns model parameters as single INDArrayINDArray
updaterState()
This method returns updater state (if applicable), null otherwise
-
-
-
Method Detail
-
init
void init()
This method does initialization of model PLEASE NOTE: All implementations should track own state, to avoid double spending
-
params
INDArray params()
This method returns model parameters as single INDArray- Returns:
-
updaterState
INDArray updaterState()
This method returns updater state (if applicable), null otherwise- Returns:
-
getOptimizer
ConvexOptimizer getOptimizer()
This method returns Optimizer used for training- Returns:
-
fit
void fit(DataSet dataSet)
This method fits model with a given DataSet- Parameters:
dataSet
-
-
fit
void fit(MultiDataSet dataSet)
This method fits model with a given MultiDataSet- Parameters:
dataSet
-
-
fit
void fit(DataSetIterator iterator)
This method fits model with a given DataSetIterator- Parameters:
iterator
-
-
fit
void fit(MultiDataSetIterator iterator)
This method fits model with a given MultiDataSetIterator- Parameters:
iterator
-
-
doEvaluation
<T extends IEvaluation> T[] doEvaluation(DataSetIterator iterator, T... evaluations)
This method executes evaluation of the model against given iterator and evaluation implementations- Parameters:
iterator
-
-
doEvaluation
<T extends IEvaluation> T[] doEvaluation(MultiDataSetIterator iterator, T... evaluations)
This method executes evaluation of the model against given iterator and evaluation implementations- Parameters:
iterator
-
-
-