TabularModel: TabularModel

Description Usage Arguments Value

View source: R/tabular_model.R

Description

Basic model for tabular data.

Usage

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
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 Oct. 25, 2021, 5:08 p.m.