public interface NeuralNetwork
Modifier and Type | Method and Description |
---|---|
<T extends IEvaluation> |
doEvaluation(DataSetIterator iterator,
T... evaluations)
This method executes evaluation of the model against given iterator and evaluation implementations
|
<T extends IEvaluation> |
doEvaluation(MultiDataSetIterator iterator,
T... evaluations)
This method executes evaluation of the model against given iterator and evaluation implementations
|
void |
fit(DataSet dataSet)
This method fits model with a given DataSet
|
void |
fit(DataSetIterator iterator)
This method fits model with a given DataSetIterator
|
void |
fit(MultiDataSet dataSet)
This method fits model with a given MultiDataSet
|
void |
fit(MultiDataSetIterator iterator)
This method fits model with a given MultiDataSetIterator
|
ConvexOptimizer |
getOptimizer()
This method returns Optimizer used for training
|
void |
init()
This method does initialization of model
PLEASE NOTE: All implementations should track own state, to avoid double spending
|
INDArray |
params()
This method returns model parameters as single INDArray
|
INDArray |
updaterState()
This method returns updater state (if applicable), null otherwise
|
void init()
INDArray params()
INDArray updaterState()
ConvexOptimizer getOptimizer()
void fit(DataSet dataSet)
dataSet
- void fit(MultiDataSet dataSet)
dataSet
- void fit(DataSetIterator iterator)
iterator
- void fit(MultiDataSetIterator iterator)
iterator
- <T extends IEvaluation> T[] doEvaluation(DataSetIterator iterator, T... evaluations)
iterator
- <T extends IEvaluation> T[] doEvaluation(MultiDataSetIterator iterator, T... evaluations)
iterator
- Copyright © 2021. All rights reserved.