Skip to content

Commit 46de9b1

Browse files
committed
fix docu
1 parent 307ab40 commit 46de9b1

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

R/LearnerTorch.R

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,9 @@
3131
#'
3232
#' @section Network Head and Target Encoding:
3333
#' Torch learners are expected to have the following output:
34-
#' * binary classification: `c(batch_size, 1)`, representing the logits for the positive class.
35-
#' * multiclass classification: `c(batch_size, n_classes)`, representing the logits for all classes.
36-
#' * regression: `c(batch_size, 1)` representing the response prediction.
34+
#' * binary classification: `(batch_size, 1)`, representing the logits for the positive class.
35+
#' * multiclass classification: `(batch_size, n_classes)`, representing the logits for all classes.
36+
#' * regression: `(batch_size, 1)` representing the response prediction.
3737
#'
3838
#' Furthermore, the target encoding is expected to be as follows:
3939
#' * regression: The `numeric` target variable of a [`TaskRegr`][mlr3::TaskRegr] is encoded as a
@@ -42,7 +42,7 @@
4242
#' [`torch_float`][torch::torch_float] with shape `(batch_size, 1)` where the positive class is `1` and the negative
4343
#' class is `0`.
4444
#' * multi-class classification: The `factor` target variable of a [`TaskClassif`][mlr3::TaskClassif] is a label-encoded
45-
#' [`torch_long`][torch::torch_long] with shape `(batch_size, n_classes)` starting at `1` and ending at `n_classes`.
45+
#' [`torch_long`][torch::torch_long] with shape `(batch_size)` where the label-encoding goes from `1` to `n_classes`.
4646
#'
4747
#' @template param_id
4848
#' @template param_task_type

man/mlr_learners_torch.Rd

Lines changed: 4 additions & 4 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)