create_head: Create_head

In fastai: Interface to 'fastai'

Create_head

Description

Model head that takes 'nf' features, runs through 'lin_ftrs', and out 'n_out' classes.

Usage

create_head(
  nf,
  n_out,
  lin_ftrs = NULL,
  ps = 0.5,
  concat_pool = TRUE,
  bn_final = FALSE,
  lin_first = FALSE,
  y_range = NULL
)

Arguments

nf	number of features
n_out	number of out features
lin_ftrs	linear features
ps	parameter server
concat_pool	concatenate pooling
bn_final	batch normalization final
lin_first	linear first
y_range	y_range

Value

None

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