Snorkel Classification PackageΒΆ
PyTorch-based multi-task learning framework for discriminative modeling.
Manager for checkpointing model. |
|
Manager for checkpointing model. |
|
A DataLoader that uses the appropriate collate_fn for a |
|
A dataset where both the data fields and labels are stored in as dictionaries. |
|
A class to manage logging during training progress. |
|
Manager for checkpointing model. |
|
A class for writing logs. |
|
Manager for checkpointing model. |
|
A classifier built from one or more tasks to support advanced workflows. |
|
A single operation (forward pass of a module) to execute in a Task. |
|
A single task (a collection of modules and specified path through them). |
|
A class for logging to Tensorboard during training process. |
|
A class for training a MultitaskClassifier. |
|
Calculate cross-entropy loss when targets are probabilities (floats), not ints. |