Skip to content

Commit 477c4bc

Browse files
committed
[SPARK-19282] update the documentation on model summary
1 parent d16b65c commit 477c4bc

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

R/pkg/R/mllib_tree.R

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -200,8 +200,8 @@ setMethod("spark.gbt", signature(data = "SparkDataFrame", formula = "formula"),
200200
#' @return \code{summary} returns summary information of the fitted model, which is a list.
201201
#' The list of components includes \code{formula} (formula),
202202
#' \code{numFeatures} (number of features), \code{features} (list of features),
203-
#' \code{featureImportances} (feature importances), \code{numTrees} (number of trees),
204-
#' and \code{treeWeights} (tree weights).
203+
#' \code{featureImportances} (feature importances), \code{maxDepth} (max depth of trees),
204+
#' \code{numTrees} (number of trees), and \code{treeWeights} (tree weights).
205205
#' @rdname spark.gbt
206206
#' @aliases summary,GBTRegressionModel-method
207207
#' @export
@@ -406,8 +406,8 @@ setMethod("spark.randomForest", signature(data = "SparkDataFrame", formula = "fo
406406
#' @return \code{summary} returns summary information of the fitted model, which is a list.
407407
#' The list of components includes \code{formula} (formula),
408408
#' \code{numFeatures} (number of features), \code{features} (list of features),
409-
#' \code{featureImportances} (feature importances), \code{numTrees} (number of trees),
410-
#' and \code{treeWeights} (tree weights).
409+
#' \code{featureImportances} (feature importances), \code{maxDepth} (max depth of trees),
410+
#' \code{numTrees} (number of trees), and \code{treeWeights} (tree weights).
411411
#' @rdname spark.randomForest
412412
#' @aliases summary,RandomForestRegressionModel-method
413413
#' @export

0 commit comments

Comments
 (0)