output_dim_for: Network Output Dimension

View source: R/utils.R

output_dim_forR Documentation

Network Output Dimension

Description

Calculates the output dimension of a neural network for a given task that is expected by mlr3torch. For classification, this is the number of classes (unless it is a binary classification task, where it is 1). For regression, it is 1.

Usage

output_dim_for(x, ...)

Arguments

x

(any)
The task.

...

(any)
Additional arguments. Not used yet.


mlr-org/mlr3torch documentation built on April 17, 2025, 8:22 p.m.