[SPARK-2692] Decision Tree API update

• Type: Improvement
• Status: Closed
• Priority: Major
• Resolution: Duplicate
• Affects Version/s: 1.0.0
• Component/s: MLlib
• Target Version/s: 1.2.0
• Assignee: Joseph K. Bradley
• Reporter: Joseph K. Bradley
• Created: 25/Jul/14 18:58
• Updated: 30/Sep/14 19:45
• Resolved: 30/Sep/14 19:45

Description

Split Decision Tree API into separate Classifier and Regressor classes.

Details:
(a) Split classes: E.g.: DecisionTree --> DecisionTreeClassifier and DecisionTreeRegressor
(b) Included print() function for human-readable model descriptions
(c) Renamed Strategy to *Params. Changed to take strings instead of special types.
(d) Made configuration classes (Impurity, QuantileStrategy) private to mllib.
(e) Changed meaning of maxDepth by 1 to match scikit-learn and rpart.
(f) Removed static train() functions in favor of using Params classes.
(g) Introduced DatasetInfo class for metadata.

Issue Links

• Is contained by: SPARK-1856 Standardize MLlib interfaces