text_classifier_learner: Text_classifier_learner

View source: R/text_learner.R

text_classifier_learnerR Documentation

Text_classifier_learner

Description

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

Usage

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 June 22, 2024, 11:15 a.m.