tune

The 'cito' package provides a user-friendly interface for training and interpreting deep neural networks (DNN). 'cito' simplifies the fitting of DNNs by supporting the familiar formula syntax, hyperparameter tuning under cross-validation, and helps to detect and handle convergence problems.  DNNs can be trained on CPU, GPU and MacOS GPUs. In addition, 'cito' has many downstream functionalities such as various explainable AI (xAI) metrics (e.g. variable importance, partial dependence plots, accumulated local effect plots, and effect estimates) to interpret trained DNNs. 'cito' optionally provides confidence intervals (and p-values) for all xAI metrics and predictions. At the same time, 'cito' is computationally efficient because it is based on the deep learning framework 'torch'. The 'torch' package is native to R, so no Python installation or other API is required for this package.

Maximilian Pichler

cito

Building and Training Neural Networks

Christian Amesöder

Florian Hartig

Armin Schenk

tune function

<dl><dt>lower</dt>
<dd>numeric, numeric vector, character, lower boundaries of tuning space</dd>
<dt>upper</dt>
<dd>numeric, numeric vector, character, upper boundaries of tuning space</dd>
<dt>fixed</dt>
<dd>character, used for multi-dimensional hyperparameters such as hidden, which dimensions should be fixed</dd>
<dt>additional</dt>
<dd>numeric, additional control parameter which sets the value of the fixed argument</dd>
<dt>values</dt>
<dd>custom values from which hyperparameters are sampled, must be a matrix for hidden layers (first column == nodes, second column == number of layers)</dd></dl>

Arguments

Tune hyperparameter — tune

<dl>

<dt>lower</dt>
<dd>numeric, numeric vector, character, lower boundaries of tuning space</dd>


<dt>upper</dt>
<dd>numeric, numeric vector, character, upper boundaries of tuning space</dd>


<dt>fixed</dt>
<dd>character, used for multi-dimensional hyperparameters such as hidden, which dimensions should be fixed</dd>


<dt>additional</dt>
<dd>numeric, additional control parameter which sets the value of the fixed argument</dd>


<dt>values</dt>
<dd>custom values from which hyperparameters are sampled, must be a matrix for hidden layers (first column == nodes, second column == number of layers)</dd>

</dl>

tune: Tune hyperparameter

Description

Usage

Arguments