sequential_module: Sequential neural network module function factory

View source: R/sequential_module.R

sequential_moduleR Documentation

Sequential neural network module function factory

Description

Sequential neural network module function factory

Usage

sequential_module(layers = 1, hidden = 20, dropout = 0.1)

Arguments

layers

[numeric(1)]
Number of hidden layers.

hidden

[numeric(1)]
Number of hidden units.

dropout

[numeric(1)]
Dropout rate.

Value

A function that returns a sequential neural network module.

Examples

if (torch::torch_is_installed()) sequential_module()

crumble documentation built on April 13, 2025, 5:10 p.m.