22 #ifndef __BROWNBOOST_QUALITY_METRIC_SET_BATCH_H__
23 #define __BROWNBOOST_QUALITY_METRIC_SET_BATCH_H__
25 #include "algorithms/algorithm_quality_metric_set_batch.h"
26 #include "algorithms/classifier/binary_confusion_matrix_batch.h"
27 #include "algorithms/boosting/brownboost_quality_metric_set_types.h"
38 namespace quality_metric_set
58 class Batch :
public algorithms::quality_metric_set::Batch
65 Batch(
bool useDefaultMetrics =
true) : algorithms::quality_metric_set::Batch(useDefaultMetrics)
67 _inputData = InputDataCollectionPtr(
new InputDataCollection());
68 if (_useDefaultMetrics)
70 initializeQualityMetrics();
72 _resultCollection = ResultCollectionPtr(
new ResultCollection());
81 ResultCollectionPtr getResultCollection()
83 return services::staticPointerCast<ResultCollection,
84 algorithms::quality_metric_set::ResultCollection>(_resultCollection);
91 InputDataCollectionPtr getInputDataCollection()
93 return services::staticPointerCast<InputDataCollection,
94 algorithms::quality_metric_set::InputDataCollection>(_inputData);
98 virtual void initializeQualityMetrics()
100 inputAlgorithms[confusionMatrix] = services::SharedPtr<classifier::quality_metric::binary_confusion_matrix::Batch<> >(
101 new classifier::quality_metric::binary_confusion_matrix::Batch<>());
102 _inputData->add(confusionMatrix, algorithms::InputPtr(
103 new classifier::quality_metric::binary_confusion_matrix::Input));
108 using interface1::Batch;
daal
Definition: algorithm_base_common.h:31
daal::algorithms::brownboost::quality_metric_set::interface1::Batch::getInputDataCollection
InputDataCollectionPtr getInputDataCollection()
Definition: brownboost_quality_metric_set_batch.h:91
daal::algorithms::brownboost::quality_metric_set::interface1::Batch::Batch
Batch(bool useDefaultMetrics=true)
Definition: brownboost_quality_metric_set_batch.h:65
daal::algorithms::brownboost::quality_metric_set::interface1::Batch::getResultCollection
ResultCollectionPtr getResultCollection()
Definition: brownboost_quality_metric_set_batch.h:81
daal::algorithms::brownboost::quality_metric_set::interface1::ResultCollection
Class that implements functionality of the collection of result objects of the quality metrics algori...
Definition: brownboost_quality_metric_set_types.h:63
daal::algorithms::brownboost::quality_metric_set::interface1::Batch
Class that represents a set of quality metrics to check the model trained with the BrownBoost algorit...
Definition: brownboost_quality_metric_set_batch.h:58
daal::algorithms::brownboost::quality_metric_set::confusionMatrix
Definition: brownboost_quality_metric_set_types.h:49
daal::services::interface1::staticPointerCast
SharedPtr< T > staticPointerCast(const SharedPtr< U > &r)
Definition: daal_shared_ptr.h:422