49 #ifndef __NEURAL_NETWORKS_TRAINING_H__
50 #define __NEURAL_NETWORKS_TRAINING_H__
52 #include "algorithms/algorithm.h"
54 #include "services/daal_defines.h"
55 #include "algorithms/neural_networks/neural_networks_types.h"
56 #include "algorithms/neural_networks/neural_networks_training_types.h"
57 #include "algorithms/neural_networks/neural_networks_training_model.h"
58 #include "algorithms/neural_networks/layers/layer.h"
67 namespace neural_networks
82 template<
typename algorithmFPType, Method method, CpuType cpu>
83 class DAAL_EXPORT BatchContainer :
public TrainingContainerIface<batch>
91 BatchContainer(daal::services::Environment::env *daalEnv);
97 services::Status compute() DAAL_C11_OVERRIDE;
98 services::Status setupCompute() DAAL_C11_OVERRIDE;
99 services::Status resetCompute() DAAL_C11_OVERRIDE;
117 template<
typename algorithmFPType = DAAL_ALGORITHM_FP_TYPE, Method method = defaultDense>
118 class Batch :
public daal::algorithms::Training<batch>
122 Batch(services::SharedPtr<optimization_solver::iterative_solver::Batch > optimizationSolver_) : parameter(optimizationSolver_)
132 Batch(
const Batch<algorithmFPType, method> &other) : parameter(other.parameter), input(other.input)
146 services::Status initialize(
const services::Collection<size_t> &sampleSize,
const training::Topology &topology)
148 ResultPtr result = getResult();
149 if (!result || !result->get(neural_networks::training::model))
151 return services::Status(services::ErrorNullModel);
153 _result->get(neural_networks::training::model)->initialize<algorithmFPType>(sampleSize, topology, parameter);
154 return services::Status();
161 ResultPtr getResult()
172 services::Status setResult(
const ResultPtr& res)
174 DAAL_CHECK(res, services::ErrorNullResult)
176 _res = _result.get();
177 return services::Status();
185 services::SharedPtr<Batch<algorithmFPType, method> > clone()
const
187 return services::SharedPtr<Batch<algorithmFPType, method> >(cloneImpl());
194 virtual int getMethod() const DAAL_C11_OVERRIDE {
return(
int) method; }
202 Training<batch>::_ac =
new __DAAL_ALGORITHM_CONTAINER(batch, BatchContainer, algorithmFPType, method)(&_env);
205 _result.reset(
new Result());
208 virtual Batch<algorithmFPType, method> *cloneImpl() const DAAL_C11_OVERRIDE
210 return new Batch<algorithmFPType, method>(*this);
213 virtual services::Status allocateResult() DAAL_C11_OVERRIDE
215 services::Status s = _result->allocate<algorithmFPType>(&input, ¶meter, (int) method);
216 _res = _result.get();
225 using interface1::Batch;
226 using interface1::BatchContainer;
daal::algorithms::neural_networks::training::interface1::Batch::input
Input input
Definition: neural_networks_training.h:196
daal::algorithms::neural_networks::training::interface1::Batch::clone
services::SharedPtr< Batch< algorithmFPType, method > > clone() const
Definition: neural_networks_training.h:185
daal::algorithms::neural_networks::training::interface1::Batch::Batch
Batch(const Batch< algorithmFPType, method > &other)
Definition: neural_networks_training.h:132
daal
Definition: algorithm_base_common.h:57
daal::algorithms::neural_networks::training::interface1::Batch::getMethod
virtual int getMethod() const DAAL_C11_OVERRIDE
Definition: neural_networks_training.h:194
daal::algorithms::neural_networks::training::interface1::Batch::Batch
Batch(services::SharedPtr< optimization_solver::iterative_solver::Batch > optimizationSolver_)
Definition: neural_networks_training.h:122
daal::algorithms::neural_networks::training::interface1::BatchContainer
Class containing methods to train neural network model using algorithmFPType precision arithmetic...
Definition: neural_networks_training.h:83
daal::algorithms::neural_networks::training::interface1::Batch
Provides methods for neural network model-based training in the batch processing mode.
Definition: neural_networks_training.h:118
daal::batch
Definition: daal_defines.h:131
daal::algorithms::neural_networks::training::interface1::Parameter
Class representing the parameters of neural network.
Definition: neural_networks_training_model.h:85
daal::algorithms::neural_networks::training::interface1::Batch::parameter
Parameter parameter
Definition: neural_networks_training.h:197
daal::algorithms::neural_networks::training::interface1::Batch::getResult
ResultPtr getResult()
Definition: neural_networks_training.h:161
daal::algorithms::neural_networks::training::interface1::Result
Provides methods to access result obtained with the compute() method of the neural network training a...
Definition: neural_networks_training_result.h:91
daal::services::ErrorNullModel
Definition: error_indexes.h:109
daal::algorithms::neural_networks::training::interface1::Batch::initialize
services::Status initialize(const services::Collection< size_t > &sampleSize, const training::Topology &topology)
Definition: neural_networks_training.h:146
daal::algorithms::neural_networks::training::model
Definition: neural_networks_training_result.h:78
daal::algorithms::neural_networks::training::interface1::Batch::setResult
services::Status setResult(const ResultPtr &res)
Definition: neural_networks_training.h:172
daal::services::ErrorNullResult
Definition: error_indexes.h:122
daal::algorithms::TrainingContainerIface
Abstract interface class that provides virtual methods to access and run implementations of the model...
Definition: training.h:76
daal::algorithms::Training
Provides methods to train models that depend on the data provided. For example, these methods enable ...
Definition: training.h:86