48 #ifndef __NAIVE_BAYES_QUALITY_METRIC_SET_BATCH_H__
49 #define __NAIVE_BAYES_QUALITY_METRIC_SET_BATCH_H__
51 #include "algorithms/algorithm_quality_metric_set_batch.h"
52 #include "algorithms/classifier/multiclass_confusion_matrix_batch.h"
53 #include "algorithms/naive_bayes/multinomial_naive_bayes_quality_metric_set_types.h"
59 namespace multinomial_naive_bayes
64 namespace quality_metric_set
83 class Batch :
public algorithms::quality_metric_set::Batch
92 Batch(
size_t nClasses = 2,
bool useDefaultMetrics =
true) :
93 algorithms::quality_metric_set::Batch(useDefaultMetrics),
96 _inputData = InputDataCollectionPtr(
new InputDataCollection());
97 if (_useDefaultMetrics)
99 initializeQualityMetrics();
101 _resultCollection = ResultCollectionPtr(
new ResultCollection());
110 ResultCollectionPtr getResultCollection()
112 return services::staticPointerCast<ResultCollection,
113 algorithms::quality_metric_set::ResultCollection>(_resultCollection);
120 InputDataCollectionPtr getInputDataCollection()
122 return services::staticPointerCast<InputDataCollection,
123 algorithms::quality_metric_set::InputDataCollection>(_inputData);
128 virtual void initializeQualityMetrics()
130 inputAlgorithms[confusionMatrix] = services::SharedPtr<classifier::quality_metric::multiclass_confusion_matrix::Batch<> >(
131 new classifier::quality_metric::multiclass_confusion_matrix::Batch<>(parameter.nClasses));
132 _inputData->add(confusionMatrix, algorithms::InputPtr(
133 new classifier::quality_metric::multiclass_confusion_matrix::Input));
138 using interface1::Batch;
daal::algorithms::multinomial_naive_bayes::quality_metric_set::interface1::Batch::parameter
Parameter parameter
Definition: multinomial_naive_bayes_quality_metric_set_batch.h:86
daal
Definition: algorithm_base_common.h:57
daal::algorithms::multinomial_naive_bayes::quality_metric_set::interface1::Parameter::nClasses
size_t nClasses
Definition: multinomial_naive_bayes_quality_metric_set_types.h:96
daal::algorithms::multinomial_naive_bayes::quality_metric_set::interface1::ResultCollection
Class that implements functionality of the collection of result objects of the quality metrics algori...
Definition: multinomial_naive_bayes_quality_metric_set_types.h:105
daal::algorithms::multinomial_naive_bayes::quality_metric_set::interface1::Batch::Batch
Batch(size_t nClasses=2, bool useDefaultMetrics=true)
Definition: multinomial_naive_bayes_quality_metric_set_batch.h:92
daal::algorithms::multinomial_naive_bayes::quality_metric_set::interface1::Batch::getInputDataCollection
InputDataCollectionPtr getInputDataCollection()
Definition: multinomial_naive_bayes_quality_metric_set_batch.h:120
daal::algorithms::multinomial_naive_bayes::quality_metric_set::interface1::Parameter
Parameters for the Naive Bayes compute() method.
Definition: multinomial_naive_bayes_quality_metric_set_types.h:91
daal::algorithms::multinomial_naive_bayes::quality_metric_set::confusionMatrix
Definition: multinomial_naive_bayes_quality_metric_set_types.h:75
daal::algorithms::multinomial_naive_bayes::quality_metric_set::interface1::Batch::getResultCollection
ResultCollectionPtr getResultCollection()
Definition: multinomial_naive_bayes_quality_metric_set_batch.h:110
daal::services::interface1::staticPointerCast
SharedPtr< T > staticPointerCast(const SharedPtr< U > &r)
Definition: daal_shared_ptr.h:448
daal::algorithms::multinomial_naive_bayes::quality_metric_set::interface1::Batch
Class containing a quality metric set to check the model trained with the Naive Bayes algorithm...
Definition: multinomial_naive_bayes_quality_metric_set_batch.h:83