You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
837 B
Python
24 lines
837 B
Python
from benchmarks.overall.methods.docling import DoclingMethod
|
|
from benchmarks.overall.methods.gt import GTMethod
|
|
from benchmarks.overall.methods.llamaparse import LlamaParseMethod
|
|
from benchmarks.overall.methods.marker import MarkerMethod
|
|
from benchmarks.overall.methods.mathpix import MathpixMethod
|
|
from benchmarks.overall.methods.mistral import MistralMethod
|
|
from benchmarks.overall.methods.olmocr import OlmOCRMethod
|
|
from benchmarks.overall.scorers.heuristic import HeuristicScorer
|
|
from benchmarks.overall.scorers.llm import LLMScorer
|
|
|
|
SCORE_REGISTRY = {
|
|
"heuristic": HeuristicScorer,
|
|
"llm": LLMScorer
|
|
}
|
|
|
|
METHOD_REGISTRY = {
|
|
"marker": MarkerMethod,
|
|
"gt": GTMethod,
|
|
"mathpix": MathpixMethod,
|
|
"llamaparse": LlamaParseMethod,
|
|
"docling": DoclingMethod,
|
|
"olmocr": OlmOCRMethod,
|
|
"mistral": MistralMethod
|
|
} |