49 #ifndef __GBT_CLASSIFICATION_PREDICT_TYPES_H__
50 #define __GBT_CLASSIFICATION_PREDICT_TYPES_H__
52 #include "algorithms/algorithm.h"
53 #include "algorithms/gradient_boosted_trees/gbt_classification_model.h"
54 #include "algorithms/classifier/classifier_predict_types.h"
62 namespace classification
97 struct DAAL_EXPORT Parameter :
public daal::algorithms::classifier::Parameter
99 Parameter(
size_t nClasses = 2) : daal::algorithms::classifier::Parameter(nClasses), nIterations(0) {}
100 Parameter(
const Parameter& o) : daal::algorithms::classifier::Parameter(o), nIterations(o.nIterations){}
109 class DAAL_EXPORT Input :
public classifier::prediction::Input
111 typedef classifier::prediction::Input super;
114 Input(
const Input& other) : super(other){}
125 data_management::NumericTablePtr
get(classifier::prediction::NumericTableInputId id)
const;
132 gbt::classification::ModelPtr
get(classifier::prediction::ModelInputId id)
const;
139 void set(classifier::prediction::NumericTableInputId
id,
const data_management::NumericTablePtr &ptr);
146 void set(classifier::prediction::ModelInputId
id,
const gbt::classification::ModelPtr &ptr);
154 services::Status check(
const daal::algorithms::Parameter *parameter,
int method)
const DAAL_C11_OVERRIDE;
158 using interface1::Parameter;
159 using interface1::Input;
166 #endif // __GBT_CLASSIFICATION_PREDICT_TYPES_H__
daal::algorithms::gbt::classification::prediction::defaultDense
Definition: gbt_classification_predict_types.h:81
daal::algorithms::gbt::classification::prediction::interface1::Parameter::nIterations
size_t nIterations
Definition: gbt_classification_predict_types.h:101
daal
Definition: algorithm_base_common.h:57
daal::algorithms::classifier::prediction::ModelInputId
ModelInputId
Definition: classifier_predict_types.h:90
daal::algorithms::em_gmm::nIterations
Definition: em_gmm_types.h:123
daal::algorithms::gbt::classification::prediction::Method
Method
Definition: gbt_classification_predict_types.h:79
daal::algorithms::classifier::prediction::NumericTableInputId
NumericTableInputId
Definition: classifier_predict_types.h:79
daal::algorithms::gbt::classification::prediction::interface1::Parameter
Parameters of the prediction algorithm.
Definition: gbt_classification_predict_types.h:97
daal::algorithms::classifier::prediction::prediction
Definition: classifier_predict_types.h:102