core.objective

Overview

Hosts a variety of PAC-Bayes training objectives that combine empirical loss and KL divergence in different ways.

Contents

  • AbstractObjective: A base interface
  • BBBObjective, FClassicObjective, FQuadObjective, McAllesterObjective, TolstikhinObjective

By choosing an appropriate objective, you can guide training to minimize a PAC-Bayes bound on the model’s risk.

 1"""
 2## Overview
 3Hosts a variety of PAC-Bayes training objectives that combine empirical
 4loss and KL divergence in different ways.
 5
 6## Contents
 7- **AbstractObjective**: A base interface
 8- **BBBObjective, FClassicObjective, FQuadObjective, McAllesterObjective, TolstikhinObjective**
 9
10By choosing an appropriate objective, you can guide training to minimize
11a PAC-Bayes bound on the model’s risk.
12"""
13
14from core.objective.AbstractObjective import AbstractObjective
15from core.objective.BBBObjective import BBBObjective
16from core.objective.FClassicObjective import FClassicObjective
17from core.objective.FQuadObjective import FQuadObjective
18from core.objective.McAllesterObjective import McAllesterObjective
19from core.objective.TolstikhinObjective import TolstikhinObjective