24 #ifndef __STUMP_CLASSIFICATION_MODEL_H__
25 #define __STUMP_CLASSIFICATION_MODEL_H__
27 #include "algorithms/algorithm.h"
28 #include "data_management/data/homogen_numeric_table.h"
29 #include "data_management/data/matrix.h"
30 #include "algorithms/classifier/classifier_model.h"
31 #include "algorithms/decision_tree/decision_tree_classification_model.h"
45 namespace classification
51 enum VariableImportanceMode
70 struct DAAL_EXPORT Parameter :
public daal::algorithms::classifier::Parameter
76 Parameter(
size_t nClasses = 2) : daal::algorithms::classifier::Parameter(nClasses),
77 splitCriterion(decision_tree::classification::gini),
78 varImportance(none) {}
79 decision_tree::classification::SplitCriterion splitCriterion;
80 VariableImportanceMode varImportance;
85 services::Status check() const DAAL_C11_OVERRIDE;
97 class DAAL_EXPORT Model : public daal::algorithms::decision_tree::classification::Model
100 DECLARE_MODEL_IFACE(Model, decision_tree::classification::Model);
110 static services::SharedPtr<Model> create(
size_t nFeatures = 0,
size_t nClasses = 2, services::Status *stat = NULL);
123 size_t getSplitFeature()
const;
129 template<
typename modelFPType>
130 DAAL_EXPORT modelFPType getSplitValue();
136 template<
typename modelFPType>
137 DAAL_EXPORT modelFPType getLeftValue();
143 template<
typename modelFPType>
144 DAAL_EXPORT modelFPType getRightValue();
147 Model(
size_t nFeatures,
size_t nClasses, services::Status &st);
149 services::Status serializeImpl(data_management::InputDataArchive * arch) DAAL_C11_OVERRIDE;
151 services::Status deserializeImpl(
const data_management::OutputDataArchive * arch) DAAL_C11_OVERRIDE;
158 typedef services::SharedPtr<Model> ModelPtr;
159 typedef services::SharedPtr<const Model> ModelConstPtr;
163 using interface1::Parameter;
164 using interface1::Model;
165 using interface1::ModelPtr;
166 using interface1::ModelConstPtr;
daal::algorithms::decision_forest::training::VariableImportanceMode
VariableImportanceMode
Variable importance computation mode.
Definition: decision_forest_training_parameter.h:60
daal::algorithms::decision_tree::classification::SplitCriterion
SplitCriterion
Split criterion for Decision tree classification algorithm.
Definition: decision_tree_classification_model.h:62
daal::algorithms::stump::classification::interface1::Parameter::Parameter
Parameter(size_t nClasses=2)
Definition: stump_classification_model.h:76
daal::algorithms::stump::classification::interface1::Parameter::splitCriterion
decision_tree::classification::SplitCriterion splitCriterion
Definition: stump_classification_model.h:79
daal::algorithms::stump::classification::interface1::Parameter::varImportance
VariableImportanceMode varImportance
Definition: stump_classification_model.h:80
daal::algorithms::stump::classification::interface1::Parameter
Stump algorithm parameters.
Definition: stump_classification_model.h:70
daal::algorithms::stump::classification::interface1::Model
Model of the classifier trained by the stump::classification::training::Batch algorithm.
Definition: stump_classification_model.h:97