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)