23 #ifndef __NAIVE_BAYES_TRAINING_ONLINE_H__
24 #define __NAIVE_BAYES_TRAINING_ONLINE_H__
26 #include "algorithms/algorithm.h"
27 #include "multinomial_naive_bayes_training_types.h"
28 #include "algorithms/classifier/classifier_training_online.h"
34 namespace multinomial_naive_bayes
53 template<
typename algorithmFPType, Method method, CpuType cpu>
54 class DAAL_EXPORT OnlineContainer :
public TrainingContainerIface<online>
62 OnlineContainer(daal::services::Environment::env *daalEnv);
72 services::Status compute() DAAL_C11_OVERRIDE;
79 services::Status finalizeCompute() DAAL_C11_OVERRIDE;
95 template<
typename algorithmFPType = DAAL_ALGORITHM_FP_TYPE, Method method = defaultDense>
96 class DAAL_EXPORT Online :
public classifier::training::Online
99 typedef classifier::training::Online super;
101 typedef typename super::InputType InputType;
102 typedef algorithms::multinomial_naive_bayes::Parameter ParameterType;
103 typedef algorithms::multinomial_naive_bayes::training::Result ResultType;
104 typedef algorithms::multinomial_naive_bayes::training::PartialResult PartialResultType;
110 Online(
size_t nClasses) : parameter(nClasses)
121 Online(
const Online<algorithmFPType, method> &other) :
122 classifier::training::Online(other), parameter(other.parameter)
133 virtual int getMethod() const DAAL_C11_OVERRIDE {
return(
int)method; }
139 ResultPtr getResult()
141 return services::staticPointerCast<ResultType, classifier::training::Result>(_result);
150 services::Status setResult(
const ResultPtr& result)
152 DAAL_CHECK(result, services::ErrorNullResult)
154 _res = _result.get();
155 return services::Status();
163 _result.reset(
new ResultType());
171 PartialResultPtr getPartialResult() {
return PartialResultType::cast(_partialResult); }
178 services::SharedPtr<Online<algorithmFPType, method> > clone()
const
180 return services::SharedPtr<Online<algorithmFPType, method> >(cloneImpl());
183 ParameterType parameter;
187 virtual Online<algorithmFPType, method> * cloneImpl() const DAAL_C11_OVERRIDE
189 return new Online<algorithmFPType, method>(*this);
192 services::Status allocateResult() DAAL_C11_OVERRIDE
194 PartialResultPtr pres = getPartialResult();
195 ResultPtr res = services::staticPointerCast<ResultType, classifier::training::Result>(_result);
196 services::Status s = res->template allocate<algorithmFPType>(pres.get(), ¶meter, (int)method);
197 _res = _result.get();
201 services::Status allocatePartialResult() DAAL_C11_OVERRIDE
203 PartialResultPtr pres = getPartialResult();
204 services::Status s = pres->template allocate<algorithmFPType>((classifier::training::InputIface *)(&input), ¶meter, (int)method);
205 _pres = _partialResult.get();
209 services::Status initializePartialResult() DAAL_C11_OVERRIDE
211 PartialResultPtr pres = getPartialResult();
212 services::Status s = pres->template initialize<algorithmFPType>((classifier::training::InputIface *)(&input), ¶meter, (int)method);
213 _pres = _partialResult.get();
219 _ac =
new __DAAL_ALGORITHM_CONTAINER(online, OnlineContainer, algorithmFPType, method)(&_env);
221 _result.reset(
new ResultType());
222 _partialResult.reset(
new PartialResultType());
227 using interface1::OnlineContainer;
228 using interface1::Online;
daal::algorithms::multinomial_naive_bayes::training::interface1::Online::parameter
ParameterType parameter
Definition: multinomial_naive_bayes_training_online.h:183
daal
Definition: algorithm_base_common.h:31
daal::algorithms::multinomial_naive_bayes::training::interface1::OnlineContainer
Class containing computation methods for naive Bayes training in the online processing mode...
Definition: multinomial_naive_bayes_training_online.h:54
daal::algorithms::multinomial_naive_bayes::training::interface1::Online::Online
Online(const Online< algorithmFPType, method > &other)
Definition: multinomial_naive_bayes_training_online.h:121
daal::algorithms::multinomial_naive_bayes::training::interface1::Online::Online
Online(size_t nClasses)
Definition: multinomial_naive_bayes_training_online.h:110
daal::algorithms::multinomial_naive_bayes::training::interface1::Online::resetResult
void resetResult()
Definition: multinomial_naive_bayes_training_online.h:161
daal::algorithms::multinomial_naive_bayes::training::interface1::Online::getPartialResult
PartialResultPtr getPartialResult()
Definition: multinomial_naive_bayes_training_online.h:171
daal::algorithms::multinomial_naive_bayes::training::interface1::Online::getMethod
virtual int getMethod() const DAAL_C11_OVERRIDE
Definition: multinomial_naive_bayes_training_online.h:133
daal::online
Definition: daal_defines.h:112
daal::algorithms::multinomial_naive_bayes::training::interface1::Online::getResult
ResultPtr getResult()
Definition: multinomial_naive_bayes_training_online.h:139
daal::algorithms::multinomial_naive_bayes::training::interface1::Online::clone
services::SharedPtr< Online< algorithmFPType, method > > clone() const
Definition: multinomial_naive_bayes_training_online.h:178
daal::services::ErrorNullResult
Definition: error_indexes.h:96
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::multinomial_naive_bayes::training::interface1::Online::setResult
services::Status setResult(const ResultPtr &result)
Definition: multinomial_naive_bayes_training_online.h:150