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>
121 typedef algorithms::neural_networks::training::Input InputType;
122 typedef algorithms::neural_networks::training::Parameter ParameterType;
123 typedef algorithms::neural_networks::training::Result ResultType;
126 Batch(services::SharedPtr<optimization_solver::iterative_solver::Batch > optimizationSolver_) : parameter(optimizationSolver_)
136 Batch(
const Batch<algorithmFPType, method> &other) : parameter(other.parameter), input(other.input)
150 services::Status initialize(
const services::Collection<size_t> &sampleSize,
const training::Topology &topology)
152 ResultPtr result = getResult();
153 if (!result || !result->get(neural_networks::training::model))
155 return services::Status(services::ErrorNullModel);
157 _result->get(neural_networks::training::model)->initialize<algorithmFPType>(sampleSize, topology, parameter);
158 return services::Status();
165 ResultPtr getResult()
176 services::Status setResult(
const ResultPtr& res)
178 DAAL_CHECK(res, services::ErrorNullResult)
180 _res = _result.get();
181 return services::Status();
189 services::SharedPtr<Batch<algorithmFPType, method> > clone()
const 191 return services::SharedPtr<Batch<algorithmFPType, method> >(cloneImpl());
198 virtual int getMethod() const DAAL_C11_OVERRIDE {
return(
int) method; }
201 ParameterType parameter;
206 Training<batch>::_ac =
new __DAAL_ALGORITHM_CONTAINER(batch, BatchContainer, algorithmFPType, method)(&_env);
209 _result.reset(
new ResultType());
212 virtual Batch<algorithmFPType, method> *cloneImpl()
const DAAL_C11_OVERRIDE
214 return new Batch<algorithmFPType, method>(*this);
217 virtual services::Status allocateResult() DAAL_C11_OVERRIDE
219 services::Status s = _result->allocate<algorithmFPType>(&input, ¶meter, (int) method);
220 _res = _result.get();
229 using interface1::Batch;
230 using interface1::BatchContainer;
daal::algorithms::neural_networks::training::interface1::Topology
Class defining a neural network topology - a set of layers and connection between them - on the train...
Definition: neural_networks_training_topology.h:66
daal::algorithms::neural_networks::training::interface1::Batch::Batch
Batch(const Batch< algorithmFPType, method > &other)
Definition: neural_networks_training.h:136
daal::services::interface1::Environment::_envStruct
The environment structure.
Definition: env_detect.h:95
daal::services::interface1::Status
Class that holds the results of API calls. In case of API routine failure it contains the list of err...
Definition: error_handling.h:491
daal
Definition: algorithm_base_common.h:57
daal::algorithms::neural_networks::training::interface1::Batch::clone
services::SharedPtr< Batch< algorithmFPType, method > > clone() const
Definition: neural_networks_training.h:189
daal::algorithms::neural_networks::training::interface1::Batch::parameter
ParameterType parameter
Definition: neural_networks_training.h:201
daal::algorithms::neural_networks::training::interface1::Batch::getMethod
virtual int getMethod() const DAAL_C11_OVERRIDE
Definition: neural_networks_training.h:198
daal::algorithms::neural_networks::training::interface1::Batch::Batch
Batch(services::SharedPtr< optimization_solver::iterative_solver::Batch > optimizationSolver_)
Definition: neural_networks_training.h:126
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:132
daal::services::interface1::SharedPtr
Shared pointer that retains shared ownership of an object through a pointer. Several SharedPtr object...
Definition: daal_shared_ptr.h:187
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::getResult
ResultPtr getResult()
Definition: neural_networks_training.h:165
daal::services::interface1::SharedPtr::get
T * get() const
Definition: daal_shared_ptr.h:332
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::algorithms::neural_networks::training::interface1::Batch::input
InputType input
Definition: neural_networks_training.h:200
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:150
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:176
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
daal::services::interface1::Collection
Class that implements functionality of the Collection container.
Definition: collection.h:69