From 1582ce8ba02cf5e905931ccf01a7acc70275a3d4 Mon Sep 17 00:00:00 2001 From: Bharat Lal Date: Mon, 9 Nov 2015 12:18:29 +0530 Subject: [PATCH] corrected computation of MSE in mllib docs --- docs/mllib-decision-tree.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/mllib-decision-tree.md b/docs/mllib-decision-tree.md index f31c4f88936b..b5b454bc6924 100644 --- a/docs/mllib-decision-tree.md +++ b/docs/mllib-decision-tree.md @@ -439,7 +439,7 @@ Double testMSE = public Double call(Double a, Double b) { return a + b; } - }) / data.count(); + }) / testData.count(); System.out.println("Test Mean Squared Error: " + testMSE); System.out.println("Learned regression tree model:\n" + model.toDebugString());