nemoguardrails.evaluate.utils

View as Markdown

Module Contents

Functions

NameDescription
initialize_llmInitializes the model from LLM provider.
load_datasetLoads a dataset from a file.

API

nemoguardrails.evaluate.utils.initialize_llm(
model_config: nemoguardrails.rails.llm.config.Model
)

Initializes the model from LLM provider.

nemoguardrails.evaluate.utils.load_dataset(
dataset_path: str
)

Loads a dataset from a file.