Closed smurching closed 7 years ago
This PR looks kinda big since I alphabetized the configs in mllib-small.yaml (to help keep track of which tests were being run). Let me know if it'd be helpful for me to de-alphabetize them/just commit the new test configs I've added.
LGTM I ran it locally Merging with master Thanks @smurching and @WeichenXu123 !
This PR follows up on #112, adding new performance tests for DecisionTreeRegression, RandomForestRegression, GMM, and HashingTF.
Summary of changes:
TreeOrForestClassification
into a trait (TreeOrForestEstimator
) exposing methods for all tree/forest estimator performance tests.DecisionTreeClassification.scala
intoTreeOrForestEstimator.scala
I tested this PR by running the performance tests specified in
mllib-small.yaml