create_tree: AML CART

In jmwerner/ArtisanalMachineLearning: Hand Made Pure Organic Machine Learning

Description

Produces a single decision tree.

Usage

create_tree(data, response, evaluation_criterion = sum_of_squares,
  min_obs = 5, max_depth = 8)

Arguments

data	Input data.frame dimension n x p for training decision tree
response	Response vector of size nx1 corresponding to the training data
evaluation_criterion	Function that calculates error criterion for fitting, defaults to sum of squares
min_obs	Minimum observations allowed to end up in a single node, defaults to 5
max_depth	Maximum number of successive splits allowed to happen in the tree, defaults to 8

Value

Single decision tree in list format of class aml_tree

jmwerner/ArtisanalMachineLearning documentation built on Jan. 7, 2020, 1:50 a.m.