22 #ifndef __MULTI_CLASS_CLASSIFIER_QUALITY_METRIC_SET_BATCH_H__
23 #define __MULTI_CLASS_CLASSIFIER_QUALITY_METRIC_SET_BATCH_H__
25 #include "algorithms/algorithm_quality_metric_set_batch.h"
26 #include "algorithms/classifier/multiclass_confusion_matrix_batch.h"
27 #include "algorithms/multi_class_classifier/multi_class_classifier_quality_metric_set_types.h"
33 namespace multi_class_classifier
38 namespace quality_metric_set
58 class Batch :
public algorithms::quality_metric_set::Batch
67 Batch(
size_t nClasses = 2,
bool useDefaultMetrics =
true) :
68 algorithms::quality_metric_set::Batch(useDefaultMetrics),
71 _inputData = InputDataCollectionPtr(
new InputDataCollection());
72 if (_useDefaultMetrics)
74 initializeQualityMetrics();
76 _resultCollection = ResultCollectionPtr(
new ResultCollection());
85 ResultCollectionPtr getResultCollection()
87 return services::staticPointerCast<ResultCollection,
88 algorithms::quality_metric_set::ResultCollection>(_resultCollection);
95 InputDataCollectionPtr getInputDataCollection()
97 return services::staticPointerCast<InputDataCollection,
98 algorithms::quality_metric_set::InputDataCollection>(_inputData);
102 virtual void initializeQualityMetrics()
104 inputAlgorithms[confusionMatrix] = services::SharedPtr<classifier::quality_metric::multiclass_confusion_matrix::Batch<> >(
105 new classifier::quality_metric::multiclass_confusion_matrix::Batch<>(parameter.nClasses));
106 _inputData->add(confusionMatrix, algorithms::InputPtr(
107 new classifier::quality_metric::multiclass_confusion_matrix::Input));
112 using interface1::Batch;
daal::algorithms::multi_class_classifier::quality_metric_set::confusionMatrix
Definition: multi_class_classifier_quality_metric_set_types.h:52
daal
Definition: algorithm_base_common.h:31
daal::algorithms::multi_class_classifier::quality_metric_set::interface1::Batch::Batch
Batch(size_t nClasses=2, bool useDefaultMetrics=true)
Definition: multi_class_classifier_quality_metric_set_batch.h:67
daal::algorithms::multi_class_classifier::quality_metric_set::interface1::Parameter::nClasses
size_t nClasses
Definition: multi_class_classifier_quality_metric_set_types.h:73
daal::algorithms::multi_class_classifier::quality_metric_set::interface1::Parameter
Parameters for the multi-class classifier compute() method.
Definition: multi_class_classifier_quality_metric_set_types.h:68
daal::algorithms::multi_class_classifier::quality_metric_set::interface1::Batch::getResultCollection
ResultCollectionPtr getResultCollection()
Definition: multi_class_classifier_quality_metric_set_batch.h:85
daal::algorithms::multi_class_classifier::quality_metric_set::interface1::Batch::getInputDataCollection
InputDataCollectionPtr getInputDataCollection()
Definition: multi_class_classifier_quality_metric_set_batch.h:95
daal::algorithms::multi_class_classifier::quality_metric_set::interface1::Batch
Class containing a set of quality metrics to check the model trained with the multi-class classifier ...
Definition: multi_class_classifier_quality_metric_set_batch.h:58
daal::algorithms::multi_class_classifier::quality_metric_set::interface1::Batch::parameter
Parameter parameter
Definition: multi_class_classifier_quality_metric_set_batch.h:61
daal::algorithms::multi_class_classifier::quality_metric_set::interface1::ResultCollection
Class that implements functionality of the collection of result objects of the quality metrics algori...
Definition: multi_class_classifier_quality_metric_set_types.h:82
daal::services::interface1::staticPointerCast
SharedPtr< T > staticPointerCast(const SharedPtr< U > &r)
Definition: daal_shared_ptr.h:422