IAI wrappers
Work in progress...
Main.OCTHaGOn.find_leaves — Functionfind_leaves(lnr::OptimalTreeLearner)Finds all leaves of OptimalTreeLearner.
Main.OCTHaGOn.pwl_constraint_data — Functionpwl_constraint_data(lnr::IAI.OptimalTreeLearner, vks)Returns the regression weights from an OptimalTreeLearner by leaf. Arguments:
- lnr: OptimalTreeLearner
- vks: headers of DataFrame X, i.e. varkeys
Returns Dict[leaf_number] containing [B0, B].
Main.OCTHaGOn.trust_region_data — Functiontrust_region_data(lnr:: IAI.OptimalTreeLearner, vks)Returns the hyperplane data from an OptimalTreeLearner. Arguments:
- lnr: OptimalTreeLearner
- vks: headers of DataFrame X, i.e. varkeys
Returns Dict[leaf_number] containing [B0, B].
Main.OCTHaGOn.check_if_trained — FunctionChecks if a learner is trained.