tyrin/doc-extract
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

(no commit message)

Browse Source
This commit is contained in:
Tyrin Todd
2026-04-02 18:15:47 -07:00
parent 2b40f813a0
commit 051376c286
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
optimize.py
View File

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

3
program.py
View File

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