Configs¶

dnallm.configuration.configs ¶

Classes¶

BenchmarkConfig ¶

Bases: BaseModel

Top-level configuration for the DNA Language Model benchmark. This class validates and structures the entire YAML configuration file, where each top-level key in the YAML corresponds to an attribute of this class.

BenchmarkInfoConfig ¶

Bases: BaseModel

Configuration for the benchmark's metadata.

DatasetConfig ¶

Bases: BaseModel

Configuration for a single dataset used in the benchmark.

EvaluationConfig ¶

Bases: BaseModel

Configuration for the evaluation phase of the benchmark.

HeadConfig ¶

Bases: BaseModel

Configuration for the classification/regression head

InferenceConfig ¶

Bases: BaseModel

Configuration for model inference

LoraConfig ¶

Bases: BaseModel

Configuration for LoRA (Low-Rank Adaptation)

ModelConfig ¶

Bases: BaseModel

Configuration for a single model to be benchmarked.

OutputConfig ¶

Bases: BaseModel

Configuration for generating benchmark reports and artifacts.

TaskConfig ¶

Bases: BaseModel

Configuration for different fine-tuning tasks

TrainingConfig ¶

Bases: BaseModel

Configuration for training

Functions¶

load_config ¶

load_config(config_path)

Load configuration from a YAML file and return a dictionary of configuration objects. Args: config_path (str): Path to the YAML configuration file.