graphdoc.train.optimizers module

graphdoc.train.optimizers.optimizer_class(optimizer_type: str, optimizer_kwargs: Dict[str, Any])

graphdoc.train.optimizers.optimizer_compile(optimizer_type: str, optimizer_kwargs: Dict[str, Any])

Compiles the optimizer given the optimizer type and optimizer kwargs.

Optimizer kwargs are optimizer specific, and must include a student field that maps to a dspy.ChainOfThought, dspy.Predict, etc.