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 implementationsvoidfit(DataSet dataSet)This method fits model with a given DataSetvoidfit(DataSetIterator iterator)This method fits model with a given DataSetIteratorvoidfit(MultiDataSetIterator iterator)This method fits model with a given MultiDataSetIteratorvoidfit(MultiDataSet dataSet)This method fits model with a given MultiDataSetConvexOptimizergetOptimizer()This method returns Optimizer used for trainingvoidinit()This method does initialization of model PLEASE NOTE: All implementations should track own state, to avoid double spendingINDArrayparams()This method returns model parameters as single INDArrayINDArrayupdaterState()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-
-
-