Snorkel Classification PackageΒΆ

PyTorch-based multi-task learning framework for discriminative modeling.

Checkpointer

Manager for checkpointing model.

CheckpointerConfig

Manager for checkpointing model.

DictDataLoader

A DataLoader that uses the appropriate collate_fn for a DictDataset.

DictDataset

A dataset where both the data fields and labels are stored in as dictionaries.

LogManager

A class to manage logging during training progress.

LogManagerConfig

Manager for checkpointing model.

LogWriter

A class for writing logs.

LogWriterConfig

Manager for checkpointing model.

MultitaskClassifier

A classifier built from one or more tasks to support advanced workflows.

Operation

A single operation (forward pass of a module) to execute in a Task.

Task

A single task (a collection of modules and specified path through them).

TensorBoardWriter

A class for logging to Tensorboard during training process.

Trainer

A class for training a MultitaskClassifier.

cross_entropy_with_probs

Calculate cross-entropy loss when targets are probabilities (floats), not ints.