capreolus.trainer
¶
Submodules¶
Package Contents¶
Classes¶
Base class for Trainer modules. The purpose of a Trainer is to train a |
Attributes¶
- class capreolus.trainer.Trainer(config=None, provide=None, share_dependency_objects=False, build=True)[source]¶
Bases:
capreolus.ModuleBase
Base class for Trainer modules. The purpose of a Trainer is to train a
Reranker
module and use it to make predictions. Capreolus provides two trainers:PytorchTrainer
andTensorFlowTrainer
- Modules should provide:
a
train
method that trains a reranker on training and dev (validation) dataa
predict
method that uses a reranker to make predictions on data