22 #ifndef __NAIVE_BAYES_TRAINING_TYPES_H__
23 #define __NAIVE_BAYES_TRAINING_TYPES_H__
25 #include "algorithms/naive_bayes/multinomial_naive_bayes_model.h"
26 #include "data_management/data/data_collection.h"
27 #include "algorithms/classifier/classifier_training_types.h"
33 namespace multinomial_naive_bayes
60 enum Step2MasterInputId
63 lastStep2MasterInputId = partialModels
78 class DAAL_EXPORT PartialResult :
public classifier::training::PartialResult
81 DECLARE_SERIALIZABLE_CAST(PartialResult);
84 virtual ~PartialResult() {}
91 multinomial_naive_bayes::PartialModelPtr
get(classifier::training::PartialResultId id)
const;
101 template <
typename algorithmFPType>
102 DAAL_EXPORT services::Status allocate(
const daal::algorithms::Input *input,
const daal::algorithms::Parameter *parameter,
const int method);
112 template <
typename algorithmFPType>
113 DAAL_EXPORT services::Status initialize(
const daal::algorithms::Input *input,
const daal::algorithms::Parameter *parameter,
const int method);
119 size_t getNumberOfFeatures()
const;
129 services::Status check(
const daal::algorithms::Input *input,
const daal::algorithms::Parameter *parameter,
int method)
const DAAL_C11_OVERRIDE;
138 services::Status check(
const daal::algorithms::Parameter *parameter,
int method)
const DAAL_C11_OVERRIDE;
142 template<
typename Archive,
bool onDeserialize>
143 services::Status serialImpl(Archive *arch)
145 return classifier::training::PartialResult::serialImpl<Archive, onDeserialize>(arch);
148 services::Status checkImpl(
size_t nFeatures,
const daal::algorithms::Parameter* parameter)
const;
150 typedef services::SharedPtr<PartialResult> PartialResultPtr;
159 class DAAL_EXPORT Result :
public classifier::training::Result
162 DECLARE_SERIALIZABLE_CAST(Result);
171 multinomial_naive_bayes::ModelPtr
get(classifier::training::ResultId id)
const;
181 template <
typename algorithmFPType>
182 DAAL_EXPORT services::Status allocate(
const daal::algorithms::Input *input,
const daal::algorithms::Parameter *parameter,
const int method);
192 template <
typename algorithmFPType>
193 DAAL_EXPORT services::Status allocate(
const daal::algorithms::PartialResult *partialResult,
const daal::algorithms::Parameter *parameter,
const int method);
203 services::Status check(
const daal::algorithms::PartialResult *partialResult,
const daal::algorithms::Parameter *parameter,
int method)
const DAAL_C11_OVERRIDE;
213 services::Status check(
const daal::algorithms::Input *input,
const daal::algorithms::Parameter *parameter,
int method)
const DAAL_C11_OVERRIDE;
217 template<
typename Archive,
bool onDeserialize>
218 services::Status serialImpl(Archive *arch)
220 return classifier::training::Result::serialImpl<Archive, onDeserialize>(arch);
223 services::Status checkImpl(
size_t nFeatures,
const daal::algorithms::Parameter* parameter)
const;
225 typedef services::SharedPtr<Result> ResultPtr;
231 class DAAL_EXPORT DistributedInput :
public classifier::training::InputIface
235 DistributedInput(
const DistributedInput& other) : classifier::training::InputIface(other){}
237 virtual ~DistributedInput() {}
239 virtual size_t getNumberOfFeatures()
const DAAL_C11_OVERRIDE;
246 data_management::DataCollectionPtr
get(Step2MasterInputId id)
const;
253 void add(
const Step2MasterInputId &
id,
const PartialResultPtr &partialResult);
260 void set(Step2MasterInputId
id,
const data_management::DataCollectionPtr &value);
267 services::Status check(
const daal::algorithms::Parameter *parameter,
int method)
const DAAL_C11_OVERRIDE;
271 using interface1::DistributedInput;
272 using interface1::PartialResult;
273 using interface1::PartialResultPtr;
274 using interface1::Result;
275 using interface1::ResultPtr;
daal
Definition: algorithm_base_common.h:31
daal::algorithms::multinomial_naive_bayes::training::defaultDense
Definition: multinomial_naive_bayes_training_types.h:52
daal::algorithms::multinomial_naive_bayes::training::Step2MasterInputId
Step2MasterInputId
Definition: multinomial_naive_bayes_training_types.h:60
daal::algorithms::multinomial_naive_bayes::training::fastCSR
Definition: multinomial_naive_bayes_training_types.h:53
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:78
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:159
daal::algorithms::classifier::training::ResultId
ResultId
Definition: classifier_training_types.h:80
daal::algorithms::math::abs::value
Definition: abs_types.h:86
daal::algorithms::multinomial_naive_bayes::training::partialModels
Definition: multinomial_naive_bayes_training_types.h:62
daal::algorithms::classifier::training::PartialResultId
PartialResultId
Definition: classifier_training_types.h:70
daal::algorithms::multinomial_naive_bayes::training::Method
Method
Definition: multinomial_naive_bayes_training_types.h:50