The ClassificationConfig class provides configuration options for supervised fine-tuning (SFT) of language models using adapter-based approaches. This configuration class is essential when creating new adapters as it defines all the training parameters and hyperparameters needed for the fine-tuning process.

​
General Hyperparameters

ParameterTypeRequiredDefaultDescription
base_modelstringYesThe base model to fine-tune
adapterstringNoloraThe type of adapter to use. Only lora is allowed.
epochsintegerNo3Number of epochs to train for
train_stepsintegerNo-Number of training steps (overrides epochs if set)
learning_ratefloatNo0.0002Learning rate for training
enable_early_stoppingbooleanNotrueWhether to enable early stopping
lr_schedulerobjectNocosine_with_restartsLearning rate scheduler configuration
optimizerobjectNoadamw_torchOptimizer configuration
warmup_ratiofloatNo0.03Ratio of training steps to use for warmup
effective_batch_sizeintegerNo16Effective batch size for training

​
LoRA Specific Hyperparameters

ParameterTypeRequiredDefaultDescription
rankintegerNo16Rank of the LoRA adapter
target_modulesarray[string]NoTarget modules in attention blocksList of model modules to fine-tune
lora_alphaintegerNo2 x rankAlpha parameter for LoRA
lora_dropoutfloatNo0.00Dropout rate for LoRA

​
Example Usage

from predibase import ClassificationConfig

# Basic configuration
config = ClassificationConfig(
    base_model="qwen3-8b"
)

# Advanced configuration
config = ClassificationConfig(
    base_model="qwen3-8b",
    epochs=1,
    rank=32,
    target_modules=["q_proj", "v_proj", "k_proj", "o_proj"]
)