TextLearner: TextLearner

View source: R/text_learner.R

TextLearnerR Documentation

TextLearner

Description

Basic class for a 'Learner' in NLP.

Usage

TextLearner(
  dls,
  model,
  alpha = 2,
  beta = 1,
  moms = list(0.8, 0.7, 0.8),
  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
)

Arguments

dls

dls

model

model

alpha

alpha

beta

beta

moms

moms

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

Value

None


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