22 #ifndef __GBT_CLASSIFICATION_TRAINING_TYPES_H__
23 #define __GBT_CLASSIFICATION_TRAINING_TYPES_H__
25 #include "algorithms/algorithm.h"
26 #include "algorithms/classifier/classifier_training_types.h"
27 #include "algorithms/gradient_boosted_trees/gbt_classification_model.h"
28 #include "algorithms/gradient_boosted_trees/gbt_training_parameter.h"
36 namespace classification
82 struct DAAL_EXPORT Parameter :
public classifier::Parameter,
public daal::algorithms::gbt::training::Parameter
85 Parameter(
size_t nClasses) : classifier::Parameter(nClasses), loss(crossEntropy) {}
86 services::Status check() const DAAL_C11_OVERRIDE;
87 LossFunctionType loss;
97 class DAAL_EXPORT Result : public classifier::training::Result
100 DECLARE_SERIALIZABLE_CAST(Result);
110 ModelPtr
get(classifier::training::ResultId id)
const;
117 void set(classifier::training::ResultId
id,
const ModelPtr &value);
126 template <
typename algorithmFPType>
127 DAAL_EXPORT services::Status allocate(
const daal::algorithms::Input *input,
const daal::algorithms::Parameter *parameter,
const int method);
136 services::Status check(
const daal::algorithms::Input *input,
const daal::algorithms::Parameter *par,
int method)
const DAAL_C11_OVERRIDE;
140 template<
typename Archive,
bool onDeserialize>
141 services::Status serialImpl(Archive *arch)
143 return daal::algorithms::Result::serialImpl<Archive, onDeserialize>(arch);
146 typedef services::SharedPtr<Result> ResultPtr;
149 using interface1::Parameter;
150 using interface1::Result;
151 using interface1::ResultPtr;
159 #endif // __GBT_CLASSIFICATION_TRAINING_TYPES_H__
daal::algorithms::gbt::classification::training::defaultDense
Definition: gbt_classification_training_types.h:57
daal::algorithms::gbt::classification::training::LossFunctionType
LossFunctionType
Loss function type.
Definition: gbt_classification_training_types.h:64
daal
Definition: algorithm_base_common.h:31
daal::algorithms::gbt::classification::training::Method
Method
Computation methods for gradient boosted trees classification model-based training.
Definition: gbt_classification_training_types.h:53
daal::algorithms::gbt::classification::training::xboost
Definition: gbt_classification_training_types.h:55
daal::algorithms::gbt::classification::training::interface1::Parameter::Parameter
Parameter(size_t nClasses)
Definition: gbt_classification_training_types.h:85
daal::algorithms::gbt::classification::training::interface1::Parameter
Gradient Boosted Trees algorithm parameters.
Definition: gbt_classification_training_types.h:82
daal::algorithms::classifier::training::ResultId
ResultId
Definition: classifier_training_types.h:80
daal::algorithms::math::abs::value
Definition: abs_types.h:86
daal::algorithms::gbt::classification::training::interface1::Result
Provides methods to access the result obtained with the compute() method of model-based training...
Definition: gbt_classification_training_types.h:97