speed up tensor plot example

schalkdaniel · Mar 14, 2023 · 1c0e6e7 · 1c0e6e7
1 parent 0f3f95b
commit 1c0e6e7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/R/plot_tensor.R b/R/plot_tensor.R
@@ -19,12 +19,12 @@
 #'   A smooth surface is drawn if `nbins = NULL`.
 #' @examples
 #' cboost = Compboost$new(data = iris, target = "Petal.Length",
-#'   loss = LossQuadratic$new())
+#'   loss = LossQuadratic$new(), learning_rate = 0.2)
 #'
 #' cboost$addTensor("Sepal.Width", "Sepal.Length", df1 = 4, df2 = 4)
 #' cboost$addTensor("Sepal.Width", "Species", df1 = 4, df2 = 2)
 #'
-#' cboost$train(150L)
+#' cboost$train(50L)
 #'
 #' plotTensor(cboost, "Sepal.Width_Species_tensor")
 #' plotTensor(cboost, "Sepal.Width_Sepal.Length_tensor")

diff --git a/man/plotTensor.Rd b/man/plotTensor.Rd