[ https://issues.apache.org/jira/browse/SPARK-2692?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel ]
Xiangrui Meng updated SPARK-2692: --------------------------------- Target Version/s: 1.2.0 (was: 1.1.0) > Decision Tree API update > ------------------------ > > Key: SPARK-2692 > URL: https://issues.apache.org/jira/browse/SPARK-2692 > Project: Spark > Issue Type: Improvement > Components: MLlib > Affects Versions: 1.0.0 > Reporter: Joseph K. Bradley > Assignee: Joseph K. Bradley > > 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. -- This message was sent by Atlassian JIRA (v6.2#6252)