Skip to content
Snippets Groups Projects
Commit 1c6cf1a5 authored by Yanbo Liang's avatar Yanbo Liang Committed by Joseph K. Bradley
Browse files

[SPARK-12570][ML][DOC] DecisionTreeRegressor: provide variance of prediction: user guide update

Update user guide doc for ```DecisionTreeRegressor``` providing variance of prediction.

cc jkbradley

Author: Yanbo Liang <ybliang8@gmail.com>

Closes #10594 from yanboliang/spark-12570.
parent 6cfe341e
No related branches found
No related tags found
No related merge requests found
...@@ -535,7 +535,9 @@ The main differences between this API and the [original MLlib Decision Tree API] ...@@ -535,7 +535,9 @@ The main differences between this API and the [original MLlib Decision Tree API]
* use of DataFrame metadata to distinguish continuous and categorical features * use of DataFrame metadata to distinguish continuous and categorical features
The Pipelines API for Decision Trees offers a bit more functionality than the original API. In particular, for classification, users can get the predicted probability of each class (a.k.a. class conditional probabilities). The Pipelines API for Decision Trees offers a bit more functionality than the original API.
In particular, for classification, users can get the predicted probability of each class (a.k.a. class conditional probabilities);
for regression, users can get the biased sample variance of prediction.
Ensembles of trees (Random Forests and Gradient-Boosted Trees) are described below in the [Tree ensembles section](#tree-ensembles). Ensembles of trees (Random Forests and Gradient-Boosted Trees) are described below in the [Tree ensembles section](#tree-ensembles).
...@@ -605,6 +607,13 @@ All output columns are optional; to exclude an output column, set its correspond ...@@ -605,6 +607,13 @@ All output columns are optional; to exclude an output column, set its correspond
<td>Vector of length # classes equal to rawPrediction normalized to a multinomial distribution</td> <td>Vector of length # classes equal to rawPrediction normalized to a multinomial distribution</td>
<td>Classification only</td> <td>Classification only</td>
</tr> </tr>
<tr>
<td>varianceCol</td>
<td>Double</td>
<td></td>
<td>The biased sample variance of prediction</td>
<td>Regression only</td>
</tr>
</tbody> </tbody>
</table> </table>
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment