inst/tinytest/test-xlm-roberta.R

model <- transformer(architecture = "XLM-RoBERTa", model_name = "xlm-roberta-base")
expect_true(inherits(model, "Transformer"))
expect_true(inherits(predict(model, "testing out", type = "embed-sentence", trace = FALSE), "matrix"))

## Clean up models folder
unlink_golgotha()
bnosac/golgotha documentation built on May 28, 2020, 4:06 a.m.