Skip to content

Commit be9c846

Browse files
committed
update params so the output is shorter
1 parent fb1933d commit be9c846

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

R/pkg/vignettes/sparkr-vignettes.Rmd

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -539,7 +539,7 @@ In the following example, we use the `longley` dataset to train a random forest
539539

540540
```{r}
541541
df <- createDataFrame(longley)
542-
rfModel <- spark.randomForest(df, Employed ~ ., type = "regression", numTrees = 5)
542+
rfModel <- spark.randomForest(df, Employed ~ ., type = "regression", maxDepth = 2, numTrees = 2)
543543
summary(rfModel)
544544
predictions <- predict(rfModel, df)
545545
```
@@ -553,7 +553,7 @@ Similar to the random forest example above, we use the `longley` dataset to trai
553553

554554
```{r}
555555
df <- createDataFrame(longley)
556-
gbtModel <- spark.gbt(df, Employed ~ ., type = "regression", maxIter = 5)
556+
gbtModel <- spark.gbt(df, Employed ~ ., type = "regression", maxDepth = 2, maxIter = 2)
557557
summary(gbtModel)
558558
predictions <- predict(gbtModel, df)
559559
```

0 commit comments

Comments
 (0)