23 #ifndef __NAIVE_BAYES_TRAINING_DISTRIBUTED_H__
24 #define __NAIVE_BAYES_TRAINING_DISTRIBUTED_H__
26 #include "algorithms/algorithm.h"
27 #include "multinomial_naive_bayes_training_types.h"
33 namespace multinomial_naive_bayes
53 template<ComputeStep step,
typename algorithmFPType, Method method, CpuType cpu>
54 class DAAL_EXPORT DistributedContainer;
60 template<
typename algorithmFPType, Method method, CpuType cpu>
61 class DAAL_EXPORT DistributedContainer<step2Master, algorithmFPType, method, cpu> :
public TrainingContainerIface<distributed>
69 DistributedContainer(daal::services::Environment::env *daalEnv);
71 ~DistributedContainer();
79 services::Status compute() DAAL_C11_OVERRIDE;
86 services::Status finalizeCompute() DAAL_C11_OVERRIDE;
101 template<ComputeStep step,
typename algorithmFPType = DAAL_ALGORITHM_FP_TYPE, Method method = defaultDense>
102 class DAAL_EXPORT Distributed {};
117 template<
typename algorithmFPType, Method method>
118 class DAAL_EXPORT Distributed<step1Local, algorithmFPType, method> :
public Online<algorithmFPType, method>
121 typedef Online<algorithmFPType, method> super;
123 typedef typename super::InputType InputType;
124 typedef typename super::ParameterType ParameterType;
125 typedef typename super::ResultType ResultType;
126 typedef typename super::PartialResultType PartialResultType;
132 Distributed(
size_t nClasses) : Online<algorithmFPType, method>::Online(nClasses) {}
140 Distributed(
const Distributed<step1Local, algorithmFPType, method> &other) :
141 Online<algorithmFPType, method>(other)
149 services::SharedPtr<Distributed<step1Local, algorithmFPType, method> > clone()
const
151 return services::SharedPtr<Distributed<step1Local, algorithmFPType, method> >(cloneImpl());
155 virtual Distributed<step1Local, algorithmFPType, method> * cloneImpl() const DAAL_C11_OVERRIDE
157 return new Distributed<step1Local, algorithmFPType, method>(*this);
174 template<
typename algorithmFPType, Method method>
175 class DAAL_EXPORT Distributed<step2Master, algorithmFPType, method> :
public Training<distributed>
178 typedef algorithms::multinomial_naive_bayes::training::DistributedInput InputType;
179 typedef algorithms::multinomial_naive_bayes::Parameter ParameterType;
180 typedef algorithms::multinomial_naive_bayes::training::Result ResultType;
181 typedef algorithms::multinomial_naive_bayes::training::PartialResult PartialResultType;
183 ParameterType parameter;
190 Distributed(
size_t nClasses) : parameter(nClasses)
201 Distributed(
const Distributed<step2Master, algorithmFPType, method> &other) :
202 Training<distributed>(other), input(other.input), parameter(other.parameter)
207 virtual ~Distributed() {}
213 virtual int getMethod() const DAAL_C11_OVERRIDE {
return(
int)method; }
219 services::Status setPartialResult(
const PartialResultPtr& partialResult)
221 DAAL_CHECK(partialResult, services::ErrorNullPartialResult);
222 _partialResult = partialResult;
223 _pres = _partialResult.get();
224 return services::Status();
231 PartialResultPtr getPartialResult() {
return _partialResult; }
239 services::Status setResult(
const ResultPtr& result)
241 DAAL_CHECK(result, services::ErrorNullResult)
243 _res = _result.get();
244 return services::Status();
251 ResultPtr getResult()
253 return ResultType::cast(_result);
261 services::Status checkFinalizeComputeParams() DAAL_C11_OVERRIDE
263 PartialResultPtr partialResult = getPartialResult();
264 DAAL_CHECK(partialResult, services::ErrorNullResult);
266 DAAL_CHECK_STATUS(s, partialResult->check(_par, method));
267 ResultPtr result = getResult();
268 DAAL_CHECK(result, services::ErrorNullResult);
269 DAAL_CHECK_STATUS(s, result->check(_pres, _par, method));
278 services::SharedPtr<Distributed<step2Master, algorithmFPType, method> > clone()
const
280 return services::SharedPtr<Distributed<step2Master, algorithmFPType, method> >(cloneImpl());
284 PartialResultPtr _partialResult;
287 virtual Distributed<step2Master, algorithmFPType, method> * cloneImpl() const DAAL_C11_OVERRIDE
289 return new Distributed<step2Master, algorithmFPType, method>(*this);
292 services::Status allocateResult() DAAL_C11_OVERRIDE
294 PartialResultPtr pres = getPartialResult();
295 ResultPtr res = getResult();
296 services::Status s = res->template allocate<algorithmFPType>(pres.get(), ¶meter, (int)method);
297 _res = _result.get();
301 services::Status allocatePartialResult() DAAL_C11_OVERRIDE
303 PartialResultPtr pres = getPartialResult();
304 services::Status s = pres->template allocate<algorithmFPType>((classifier::training::InputIface *)(&input), ¶meter, (int)method);
305 _pres = _partialResult.get();
309 services::Status initializePartialResult() DAAL_C11_OVERRIDE
311 return services::Status();
316 _ac =
new __DAAL_ALGORITHM_CONTAINER(distributed, DistributedContainer, step2Master, algorithmFPType, method)(&_env);
319 _result.reset(
new ResultType());
320 _partialResult.reset(
new PartialResultType());
325 using interface1::DistributedContainer;
326 using interface1::Distributed;
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step2Master, algorithmFPType, method >::setResult
services::Status setResult(const ResultPtr &result)
Definition: multinomial_naive_bayes_training_distributed.h:239
daal
Definition: algorithm_base_common.h:31
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step1Local, algorithmFPType, method >::clone
services::SharedPtr< Distributed< step1Local, algorithmFPType, method > > clone() const
Definition: multinomial_naive_bayes_training_distributed.h:149
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step2Master, algorithmFPType, method >::getPartialResult
PartialResultPtr getPartialResult()
Definition: multinomial_naive_bayes_training_distributed.h:231
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step1Local, algorithmFPType, method >
Algorithm class for training Naive Bayes partial model in the distributed processing mode...
Definition: multinomial_naive_bayes_training_distributed.h:118
daal::services::ErrorNullPartialResult
Definition: error_indexes.h:105
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step2Master, algorithmFPType, method >::setPartialResult
services::Status setPartialResult(const PartialResultPtr &partialResult)
Definition: multinomial_naive_bayes_training_distributed.h:219
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step2Master, algorithmFPType, method >::getResult
ResultPtr getResult()
Definition: multinomial_naive_bayes_training_distributed.h:251
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step1Local, algorithmFPType, method >::Distributed
Distributed(size_t nClasses)
Definition: multinomial_naive_bayes_training_distributed.h:132
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step2Master, algorithmFPType, method >::getMethod
virtual int getMethod() const DAAL_C11_OVERRIDE
Definition: multinomial_naive_bayes_training_distributed.h:213
daal::algorithms::kmeans::init::interface1::Distributed
class DAAL_EXPORT Distributed
Computes initial clusters for the K-Means algorithm in the distributed processing mode...
Definition: kmeans_init_distributed.h:255
daal::distributed
Definition: daal_defines.h:107
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step2Master, algorithmFPType, method >::Distributed
Distributed(size_t nClasses)
Definition: multinomial_naive_bayes_training_distributed.h:190
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step2Master, algorithmFPType, method >
Algorithm class for training naive Bayes final model on the second step in the distributed processing...
Definition: multinomial_naive_bayes_training_distributed.h:175
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step2Master, algorithmFPType, method >::Distributed
Distributed(const Distributed< step2Master, algorithmFPType, method > &other)
Definition: multinomial_naive_bayes_training_distributed.h:201
daal::algorithms::multinomial_naive_bayes::training::interface1::DistributedContainer
class DAAL_EXPORT DistributedContainer
Class containing methods to compute naive Bayes training results in the distributed processing mode...
Definition: multinomial_naive_bayes_training_distributed.h:54
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step1Local, algorithmFPType, method >::Distributed
Distributed(const Distributed< step1Local, algorithmFPType, method > &other)
Definition: multinomial_naive_bayes_training_distributed.h:140
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step2Master, algorithmFPType, method >::parameter
ParameterType parameter
Definition: multinomial_naive_bayes_training_distributed.h:183
daal::step1Local
Definition: daal_defines.h:117
daal::step2Master
Definition: daal_defines.h:118
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step2Master, algorithmFPType, method >::clone
services::SharedPtr< Distributed< step2Master, algorithmFPType, method > > clone() const
Definition: multinomial_naive_bayes_training_distributed.h:278
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed
Algorithm class for training naive Bayes model in the distributed processing mode.
Definition: multinomial_naive_bayes_training_distributed.h:102
daal::services::ErrorNullResult
Definition: error_indexes.h:96
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step2Master, algorithmFPType, method >::input
InputType input
Definition: multinomial_naive_bayes_training_distributed.h:184
daal::algorithms::TrainingContainerIface
Abstract interface class that provides virtual methods to access and run implementations of the model...
Definition: training.h:50
daal::algorithms::multinomial_naive_bayes::training::interface1::Online
Algorithm class for training naive Bayes model.
Definition: multinomial_naive_bayes_training_online.h:96
daal::algorithms::Training
Provides methods to train models that depend on the data provided. For example, these methods enable ...
Definition: training.h:60
daal::algorithms::multinomial_naive_bayes::training::interface1::Distributed< step2Master, algorithmFPType, method >::checkFinalizeComputeParams
services::Status checkFinalizeComputeParams() DAAL_C11_OVERRIDE
Definition: multinomial_naive_bayes_training_distributed.h:261