Added benchmarks to mllib-large.yaml for classifcation Estimators. (#143)

This commit is contained in:
Bago Amirbekian 2018-05-30 08:18:49 -07:00 committed by Xiangrui Meng
parent 3786a8391e
commit 789a0f5b8b

View File

@ -11,3 +11,29 @@ benchmarks:
params:
depth: [5, 10]
numClasses: 4
- name: classification.GBTClassification
params:
numFeatures: 3000
depth: 5
numClasses: 4
maxIter: 10
- name: classification.RandomForestClassification
params:
depth: 10
numClasses: 4
maxIter: 200 # number of trees
- name: classification.LogisticRegression
params:
regParam: 0.01
tol: 0.0
maxIter: 20
- name: classification.LinearSVC
params:
regParam: 0.01
tol: 0
maxIter: 20
- name: classification.NaiveBayes
params:
numFeatures: 5000
numClasses: 2
smoothing: 1.0