pipefitter.estimator.DecisionTree.fit

DecisionTree.fit(table, *args, **kwargs)

Fit function for decision tree

Parameters:

*args : dicts or two-element tuples or consecutive key/value pairs, optional

The following types are allowed:
  • Dictionaries contain key/value pairs of parameters.
  • Two-element tuples must contain the name of the parameter in the first element and the value in the second element.
  • Consecutive key/value pairs are also allowed.

**kwargs : keyword arguments, optional

These keyword arguments are the same as on the constructor.

Returns:

DecisionTreeModel

Examples

>>> dtree = DecisionTree(target='Origin',
...                      inputs=['MPG_City', 'MPG_Highway', 'Length',
...                              'Weight', 'Type', 'Cylinders'],
...                      nominals = ['Type', 'Cylinders', 'Origin'])
>>> model = dtree.fit(training_data)