24 #ifndef __LOGITBOOST_QUALITY_METRIC_SET_BATCH_H__
25 #define __LOGITBOOST_QUALITY_METRIC_SET_BATCH_H__
27 #include "algorithms/algorithm_quality_metric_set_batch.h"
28 #include "algorithms/classifier/multiclass_confusion_matrix_batch.h"
29 #include "algorithms/boosting/logitboost_quality_metric_set_types.h"
40 namespace quality_metric_set
60 class Batch :
public algorithms::quality_metric_set::Batch
69 Batch(
size_t nClasses = 2,
bool useDefaultMetrics =
true) :
70 algorithms::quality_metric_set::Batch(useDefaultMetrics),
73 _inputData = InputDataCollectionPtr(
new InputDataCollection());
74 if (_useDefaultMetrics)
76 initializeQualityMetrics();
78 _resultCollection = ResultCollectionPtr(
new ResultCollection());
87 ResultCollectionPtr getResultCollection()
89 return services::staticPointerCast<ResultCollection,
90 algorithms::quality_metric_set::ResultCollection>(_resultCollection);
97 InputDataCollectionPtr getInputDataCollection()
99 return services::staticPointerCast<InputDataCollection,
100 algorithms::quality_metric_set::InputDataCollection>(_inputData);
104 virtual void initializeQualityMetrics()
106 inputAlgorithms[confusionMatrix] = services::SharedPtr<classifier::quality_metric::multiclass_confusion_matrix::Batch<> >(
107 new classifier::quality_metric::multiclass_confusion_matrix::Batch<>(parameter.nClasses));
108 _inputData->add(confusionMatrix, algorithms::InputPtr(
109 new classifier::quality_metric::multiclass_confusion_matrix::Input));
114 using interface1::Batch;
daal::algorithms::logitboost::quality_metric_set::interface1::Parameter
Parameters for the LogitBoost compute() method.
Definition: logitboost_quality_metric_set_types.h:67
daal::algorithms::logitboost::quality_metric_set::confusionMatrix
Definition: logitboost_quality_metric_set_types.h:51
daal::algorithms::logitboost::quality_metric_set::interface1::ResultCollection
Class that implements functionality of the collection of result objects of the quality metrics algori...
Definition: logitboost_quality_metric_set_types.h:81
daal::algorithms::logitboost::quality_metric_set::interface1::Batch::getResultCollection
ResultCollectionPtr getResultCollection()
Definition: logitboost_quality_metric_set_batch.h:87
daal::algorithms::logitboost::quality_metric_set::interface1::Batch
Class that represents a set of quality metrics to check the model trained with the LogitBoost trainin...
Definition: logitboost_quality_metric_set_batch.h:60
daal::services::interface1::staticPointerCast
SharedPtr< T > staticPointerCast(const SharedPtr< U > &r)
Definition: daal_shared_ptr.h:425
daal::algorithms::logitboost::quality_metric_set::interface1::InputDataCollection
Class that implements functionality of the collection of input objects of the quality metrics algorit...
Definition: logitboost_quality_metric_set_types.h:101
daal::algorithms::logitboost::quality_metric_set::interface1::Parameter::nClasses
size_t nClasses
Definition: logitboost_quality_metric_set_types.h:72
daal::algorithms::logitboost::quality_metric_set::interface1::Batch::getInputDataCollection
InputDataCollectionPtr getInputDataCollection()
Definition: logitboost_quality_metric_set_batch.h:97
daal::algorithms::logitboost::quality_metric_set::interface1::Batch::Batch
Batch(size_t nClasses=2, bool useDefaultMetrics=true)
Definition: logitboost_quality_metric_set_batch.h:69