(no commit message)

This commit is contained in:
2026-04-02 18:15:47 -07:00
parent 2b40f813a0
commit 051376c286
2 changed files with 4 additions and 2 deletions

View File

@@ -33,6 +33,7 @@ def classification_metric(
if __name__ == "__main__":
program = DocExtract.from_precompiled("tyrin/doc-extract")
dspy.configure(lm=dspy.LM("openai/gpt-4.1-mini", temperature=0.0))
optimized_program = dspy.GEPA(
@@ -40,7 +41,7 @@ if __name__ == "__main__":
reflection_lm=dspy.LM("openai/gpt-4.1-mini", temperature=1.0, max_tokens=8000),
auto="light",
).compile(
DocExtract(),
program,
trainset=load_dataset("data/train.jsonl"),
valset=load_dataset("data/val.jsonl"),
)

View File

@@ -25,4 +25,5 @@ class DocExtract(PrecompiledProgram):
return self.cot(context=context)
if __name__ == "__main__":
DocExtract().push_to_hub("tyrin/doc-extract")