TabularModel: TabularModel

View source: R/tabular_model.R

TabularModelR Documentation

TabularModel

Description

Basic model for tabular data.

Usage

TabularModel(
  emb_szs,
  n_cont,
  out_sz,
  layers,
  ps = NULL,
  embed_p = 0,
  y_range = NULL,
  use_bn = TRUE,
  bn_final = FALSE,
  bn_cont = TRUE,
  act_cls = nn()$ReLU(inplace = TRUE)
)

Arguments

emb_szs

embedding size

n_cont

number of cont

out_sz

output size

layers

layers

ps

ps

embed_p

embed proportion

y_range

y range

use_bn

use batch normalization

bn_final

batch normalization final

bn_cont

batch normalization cont

act_cls

activation

Value

None


fastai documentation built on June 22, 2024, 11:15 a.m.