evalexplorer-classify-gemma-4-e2b-sft

Classifies an international development evaluation report from its first pages into a JSON object with evaluation_approach, evaluation_type, temporality, themes and countries. One of the runs documented in baobabtech/evalexplorer-classify-experiments; code in code/.

Base model	unsloth/gemma-4-E2B-it
Method	SFT
Data	baobabtech/evalexplorer-data, config classify_codes, 1148 training documents
Hardware	gpu
Training time	33.1 min
Run report	gemma-4-e2b-sft--test

Training

LoRA SFT: r=16, alpha=16, all linear language layers, bf16, learning rate 0.0002, 2 epochs, batch 2 × 8 accumulation, loss on the answer only.

Scores

Metric	Test (134 documents)
JSON valid	1.000
Exact match (all five fields)	0.246
Mean field score	0.815
evaluation_approach accuracy	0.776
evaluation_type accuracy	0.851
temporality accuracy	0.731
themes micro F1	0.822
countries micro F1	0.862

Greedy decoding, thinking off. mean_field_score is the per-document mean of the five field scores (1/0 for the single-code fields, F1 for the lists). The model was trained on, and is scored here against, the EvalExplorer ingestion pipeline's LLM labels, which no person has reviewed, so these numbers measure agreement with that pipeline, not correctness. The run report also scores the same answers against an independent GLM-5.3-Flash relabelling the model never saw. This is an exploratory model; the intended next version is trained on the GLM labels.

Usage

from unsloth import FastLanguageModel, FastModel

loader = FastLanguageModel if "lfm" in "unsloth/gemma-4-E2B-it".lower() else FastModel
model, processor = loader.from_pretrained("unsloth/gemma-4-E2B-it", max_seq_length=8192, load_in_4bit=False)
from peft import PeftModel
model = PeftModel.from_pretrained(model, "baobabtech/evalexplorer-classify-gemma-4-e2b-sft")
loader.for_inference(model)

text = processor.apply_chat_template(prompt_messages, tokenize=False, add_generation_prompt=True,
                                     enable_thinking=False)   # prompt_messages from the dataset's `prompt` column