create_mlp: Create_mlp

View source: R/time_series_model.R

create_mlpR Documentation

Create_mlp

Description

A simple model builder to create a bunch of BatchNorm1d, Dropout and Linear layers, with “'act_fn“' activations.

Usage

create_mlp(ni, nout, linear_sizes = c(500, 500, 500))

Arguments

ni

number of input channels

nout

output shape

linear_sizes

linear output sizes

Value

model


fastai documentation built on March 31, 2023, 11:41 p.m.