main/cpp/TestSpec_8hpp_source.html

#pragma once


#include "models/TrainingSpec.hpp"


namespace ppforest2::test {


  inline TrainingSpec::Ptr classification_spec() {

    return TrainingSpec::builder(types::Mode::Classification).make();

  }


  inline TrainingSpec::Ptr regression_spec() {

    return TrainingSpec::builder(types::Mode::Regression)

        .grouping(grouping::by_cutpoint())

        .leaf(leaf::mean_response())

        .stop(stop::min_size(5))

        .make();

  }


}

TrainingSpec.hpp

ppforest2::TrainingSpec::Builder::leaf
Builder & leaf(leaf::LeafStrategy::Ptr v)
Definition TrainingSpec.hpp:156

ppforest2::TrainingSpec::Builder::make
Ptr make()
Shorthand for std::make_shared<TrainingSpec>(build()).

ppforest2::TrainingSpec::Builder::stop
Builder & stop(stop::StopRule::Ptr v)
Definition TrainingSpec.hpp:144

ppforest2::TrainingSpec::Builder::grouping
Builder & grouping(grouping::Grouping::Ptr v)
Definition TrainingSpec.hpp:152

ppforest2::TrainingSpec::builder
static Builder builder(types::Mode mode)
Create a builder for the given mode.
Definition TrainingSpec.hpp:227

ppforest2::TrainingSpec::Ptr
std::shared_ptr< TrainingSpec > Ptr
Definition TrainingSpec.hpp:45

ppforest2::grouping::by_cutpoint
Grouping::Ptr by_cutpoint()
Factory function for cutpoint-based grouping (regression).

ppforest2::leaf::mean_response
LeafStrategy::Ptr mean_response()
Factory function for mean-response leaf strategy.

ppforest2::stop::min_size
StopRule::Ptr min_size(int n)
Factory function for minimum-size stop rule.

ppforest2::test
Definition MockStop.hpp:10

ppforest2::test::regression_spec
TrainingSpec::Ptr regression_spec()
Minimal well-formed regression spec.
Definition TestSpec.hpp:32

ppforest2::test::classification_spec
TrainingSpec::Ptr classification_spec()
Default-valued classification spec.
Definition TestSpec.hpp:21

ppforest2::types::Mode::Classification
@ Classification
Definition Types.hpp:58

ppforest2::types::Mode::Regression
@ Regression
Definition Types.hpp:58