48 #ifndef __NAIVE_BAYES_TRAINING_TYPES_H__
49 #define __NAIVE_BAYES_TRAINING_TYPES_H__
51 #include "algorithms/naive_bayes/multinomial_naive_bayes_model.h"
52 #include "data_management/data/data_collection.h"
53 #include "algorithms/classifier/classifier_training_types.h"
59 namespace multinomial_naive_bayes
86 enum Step2MasterInputId
89 lastStep2MasterInputId = partialModels
104 class DAAL_EXPORT PartialResult :
public classifier::training::PartialResult
107 DECLARE_SERIALIZABLE_CAST(PartialResult);
110 virtual ~PartialResult() {}
117 multinomial_naive_bayes::PartialModelPtr
get(classifier::training::PartialResultId id)
const;
127 template <
typename algorithmFPType>
128 DAAL_EXPORT services::Status allocate(
const daal::algorithms::Input *input,
const daal::algorithms::Parameter *parameter,
const int method);
138 template <
typename algorithmFPType>
139 DAAL_EXPORT services::Status initialize(
const daal::algorithms::Input *input,
const daal::algorithms::Parameter *parameter,
const int method);
145 size_t getNumberOfFeatures()
const;
155 services::Status check(
const daal::algorithms::Input *input,
const daal::algorithms::Parameter *parameter,
int method)
const DAAL_C11_OVERRIDE;
164 services::Status check(
const daal::algorithms::Parameter *parameter,
int method)
const DAAL_C11_OVERRIDE;
168 template<
typename Archive,
bool onDeserialize>
169 services::Status serialImpl(Archive *arch)
171 return classifier::training::PartialResult::serialImpl<Archive, onDeserialize>(arch);
174 services::Status checkImpl(
size_t nFeatures,
const daal::algorithms::Parameter* parameter)
const;
176 typedef services::SharedPtr<PartialResult> PartialResultPtr;
185 class DAAL_EXPORT Result :
public classifier::training::Result
188 DECLARE_SERIALIZABLE_CAST(Result);
197 multinomial_naive_bayes::ModelPtr
get(classifier::training::ResultId id)
const;
207 template <
typename algorithmFPType>
208 DAAL_EXPORT services::Status allocate(
const daal::algorithms::Input *input,
const daal::algorithms::Parameter *parameter,
const int method);
218 template <
typename algorithmFPType>
219 DAAL_EXPORT services::Status allocate(
const daal::algorithms::PartialResult *partialResult,
const daal::algorithms::Parameter *parameter,
const int method);
229 services::Status check(
const daal::algorithms::PartialResult *partialResult,
const daal::algorithms::Parameter *parameter,
int method)
const DAAL_C11_OVERRIDE;
239 services::Status check(
const daal::algorithms::Input *input,
const daal::algorithms::Parameter *parameter,
int method)
const DAAL_C11_OVERRIDE;
243 template<
typename Archive,
bool onDeserialize>
244 services::Status serialImpl(Archive *arch)
246 return classifier::training::Result::serialImpl<Archive, onDeserialize>(arch);
249 services::Status checkImpl(
size_t nFeatures,
const daal::algorithms::Parameter* parameter)
const;
251 typedef services::SharedPtr<Result> ResultPtr;
257 class DAAL_EXPORT DistributedInput :
public classifier::training::InputIface
261 DistributedInput(
const DistributedInput& other) : classifier::training::InputIface(other){}
263 virtual ~DistributedInput() {}
265 virtual size_t getNumberOfFeatures()
const DAAL_C11_OVERRIDE;
272 data_management::DataCollectionPtr
get(Step2MasterInputId id)
const;
279 void add(
const Step2MasterInputId &
id,
const PartialResultPtr &partialResult);
286 void set(Step2MasterInputId
id,
const data_management::DataCollectionPtr &value);
293 services::Status check(
const daal::algorithms::Parameter *parameter,
int method)
const DAAL_C11_OVERRIDE;
297 using interface1::DistributedInput;
298 using interface1::PartialResult;
299 using interface1::PartialResultPtr;
300 using interface1::Result;
301 using interface1::ResultPtr;
daal
Definition: algorithm_base_common.h:57
daal::algorithms::multinomial_naive_bayes::training::defaultDense
Definition: multinomial_naive_bayes_training_types.h:78
daal::algorithms::multinomial_naive_bayes::training::Step2MasterInputId
Step2MasterInputId
Definition: multinomial_naive_bayes_training_types.h:86
daal::algorithms::multinomial_naive_bayes::training::fastCSR
Definition: multinomial_naive_bayes_training_types.h:79
daal::algorithms::multinomial_naive_bayes::training::interface1::PartialResult
Provides methods to access partial results obtained with the compute() method of the naive Bayes trai...
Definition: multinomial_naive_bayes_training_types.h:104
daal::algorithms::multinomial_naive_bayes::training::interface1::Result
Provides methods to access final results obtained with the compute() method of the naive Bayes traini...
Definition: multinomial_naive_bayes_training_types.h:185
daal::algorithms::classifier::training::ResultId
ResultId
Definition: classifier_training_types.h:106
daal::algorithms::math::abs::value
Definition: abs_types.h:112
daal::algorithms::multinomial_naive_bayes::training::partialModels
Definition: multinomial_naive_bayes_training_types.h:88
daal::algorithms::classifier::training::PartialResultId
PartialResultId
Definition: classifier_training_types.h:96
daal::algorithms::multinomial_naive_bayes::training::Method
Method
Definition: multinomial_naive_bayes_training_types.h:76