diff options
Diffstat (limited to 'examples/src/main/python/mllib/gradient_boosting_classification_example.py')
-rw-r--r-- | examples/src/main/python/mllib/gradient_boosting_classification_example.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/examples/src/main/python/mllib/gradient_boosting_classification_example.py b/examples/src/main/python/mllib/gradient_boosting_classification_example.py index b204cd1b31..65a03572be 100644 --- a/examples/src/main/python/mllib/gradient_boosting_classification_example.py +++ b/examples/src/main/python/mllib/gradient_boosting_classification_example.py @@ -43,7 +43,8 @@ if __name__ == "__main__": # Evaluate model on test instances and compute test error predictions = model.predict(testData.map(lambda x: x.features)) labelsAndPredictions = testData.map(lambda lp: lp.label).zip(predictions) - testErr = labelsAndPredictions.filter(lambda (v, p): v != p).count() / float(testData.count()) + testErr = labelsAndPredictions.filter( + lambda lp: lp[0] != lp[1]).count() / float(testData.count()) print('Test Error = ' + str(testErr)) print('Learned classification GBT model:') print(model.toDebugString()) |