public interface NeuralNetwork
Modifier and Type | Method and Description |
---|---|
<T extends IEvaluation> |
doEvaluation(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator,
T... evaluations)
This method executes evaluation of the model against given iterator and evaluation implementations
|
<T extends IEvaluation> |
doEvaluation(org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator,
T... evaluations)
This method executes evaluation of the model against given iterator and evaluation implementations
|
void |
fit(org.nd4j.linalg.dataset.api.DataSet dataSet)
This method fits model with a given DataSet
|
void |
fit(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)
This method fits model with a given DataSetIterator
|
void |
fit(org.nd4j.linalg.dataset.api.MultiDataSet dataSet)
This method fits model with a given MultiDataSet
|
void |
fit(org.nd4j.linalg.dataset.api.iterator.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
|
org.nd4j.linalg.api.ndarray.INDArray |
params()
This method returns model parameters as single INDArray
|
org.nd4j.linalg.api.ndarray.INDArray |
updaterState()
This method returns updater state (if applicable), null otherwise
|
void init()
org.nd4j.linalg.api.ndarray.INDArray params()
org.nd4j.linalg.api.ndarray.INDArray updaterState()
ConvexOptimizer getOptimizer()
void fit(org.nd4j.linalg.dataset.api.DataSet dataSet)
dataSet
- void fit(org.nd4j.linalg.dataset.api.MultiDataSet dataSet)
dataSet
- void fit(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator)
iterator
- void fit(org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator)
iterator
- <T extends IEvaluation> T[] doEvaluation(org.nd4j.linalg.dataset.api.iterator.DataSetIterator iterator, T... evaluations)
iterator
- <T extends IEvaluation> T[] doEvaluation(org.nd4j.linalg.dataset.api.iterator.MultiDataSetIterator iterator, T... evaluations)
iterator
- Copyright © 2017. All rights reserved.