text_classifier_learner: Text_classifier_learner

Description Usage Arguments Value

View source: R/text_learner.R

Description

Create a 'Learner' with a text classifier from 'dls' and 'arch'.

Usage

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
text_classifier_learner(
  dls,
  arch,
  seq_len = 72,
  config = NULL,
  backwards = FALSE,
  pretrained = TRUE,
  drop_mult = 0.5,
  n_out = NULL,
  lin_ftrs = NULL,
  ps = NULL,
  max_len = 1440,
  y_range = NULL,
  loss_func = NULL,
  opt_func = Adam(),
  lr = 0.001,
  splitter = trainable_params,
  cbs = NULL,
  metrics = NULL,
  path = NULL,
  model_dir = "models",
  wd = NULL,
  wd_bn_bias = FALSE,
  train_bn = TRUE,
  moms = list(0.95, 0.85, 0.95)
)

Arguments

dls

dls

arch

arch

seq_len

seq_len

config

config

backwards

backwards

pretrained

pretrained

drop_mult

drop_mult

n_out

n_out

lin_ftrs

lin_ftrs

ps

ps

max_len

max_len

y_range

y_range

loss_func

loss_func

opt_func

opt_func

lr

lr

splitter

splitter

cbs

cbs

metrics

metrics

path

path

model_dir

model_dir

wd

wd

wd_bn_bias

wd_bn_bias

train_bn

train_bn

moms

moms

Value

None


fastai documentation built on Oct. 25, 2021, 5:08 p.m.