Class CausalLMEvaluation
java.lang.Object
io.github.kirstenali.deepj.training.CausalLMEvaluation
Evaluation helpers for causal language models.
-
Method Summary
Modifier and TypeMethodDescriptionstatic EvaluationResultevaluate(CausalLM model, BatchSource source, int batches, int batchSize)
-
Method Details
-
evaluate
public static EvaluationResult evaluate(CausalLM model, BatchSource source, int batches, int batchSize)
-