yamle.trainers package

yamle.trainers.trainer_factory(trainer_type)

This function is used to create a trainer instance based on the trainer type.

Parameters:

trainer_type (str) – The type of trainer to create.

Return type:

Type[BaseTrainer]

Submodules

• yamle.trainers.calibration module
  • CalibrationTrainer
    • CalibrationTrainer.fit()
    • CalibrationTrainer.add_specific_args()
• yamle.trainers.ensemble module
  • BaggingTrainer
    • BaggingTrainer.fit()
  • EnsembleTrainer
    • EnsembleTrainer.fit()
    • EnsembleTrainer.add_specific_args()
• yamle.trainers.trainer module
  • BaseTrainer
    • BaseTrainer.fit()
    • BaseTrainer.interrupted
    • BaseTrainer.test()
    • BaseTrainer.calibrate()
    • BaseTrainer.fine_tune()
    • BaseTrainer.add_specific_args()