tests/testthat/train.R

# train script called by training_run in tests

library(tfruns)

FLAGS <- flags(
  flag_numeric("learning_rate", 0.01, "Learning rate")
)
rstudio/tfruns documentation built on Feb. 6, 2024, 11:29 a.m.