45 size_t treedepth,
size_t minleafsupport,
47 :
Mart(ntrees, shrinkage, nthresholds, 1 << treedepth, minleafsupport,
58 virtual std::string
name()
const {
69 std::shared_ptr<data::VerticalDataset> training_dataset);
82 virtual std::ostream &
put(std::ostream &os)
const;
Definition: dataset.cc:28
virtual std::string name() const
Returns the name of the ranker.
Definition: obliviousmart.h:58
virtual std::ostream & put(std::ostream &os) const
Prints the description of Algorithm, including its parameters.
Definition: obliviousmart.cc:39
virtual ~ObliviousMart()
Definition: obliviousmart.h:54
virtual pugi::xml_document * get_xml_model() const
Return the xml model representing the current object.
Definition: obliviousmart.cc:66
static const std::string NAME_
Definition: obliviousmart.h:62
size_t treedepth_
Definition: obliviousmart.h:73
Definition: obliviousmart.h:33
virtual std::unique_ptr< RegressionTree > fit_regressor_on_gradient(std::shared_ptr< data::VerticalDataset > training_dataset)
Fits a regression tree on the gradient given by the pseudo residuals.
Definition: obliviousmart.cc:55
ObliviousMart(size_t ntrees, double shrinkage, size_t nthresholds, size_t treedepth, size_t minleafsupport, size_t esr)
Initializes a new ObliviousLambdaMart instance with the given learning parameters.
Definition: obliviousmart.h:44
friend std::ostream & operator<<(std::ostream &os, const ObliviousMart &a)
The output stream operator.
Definition: obliviousmart.h:77