Skip to content

Commit 77e36c6

Browse files
authored
Editing docstring for Label Model (#1468)
* Editing docstring * Resolving PR edits
1 parent 11bf738 commit 77e36c6

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

snorkel/labeling/model/label_model.py

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,22 @@ class _CliqueData(NamedTuple):
8787

8888

8989
class LabelModel(nn.Module):
90-
"""A conditionally independent LabelModel to learn LF weights and assign training labels.
90+
r"""A model for learning the LF accuracies and combining their output labels.
91+
92+
This class learns a model of the labeling functions' conditional probabilities
93+
of outputting the true (unobserved) label `Y`, `P(\lf | Y)`, and uses this learned
94+
model to re-weight and combine their output labels.
95+
96+
This class is based on the approach in [Training Complex Models with Multi-Task
97+
Weak Supervision](https://arxiv.org/abs/1810.02840), published in AAAI'19. In this
98+
approach, we compute the inverse generalized covariance matrix of the junction tree
99+
of a given LF dependency graph, and perform a matrix completion-style approach with
100+
respect to these empirical statistics. The result is an estimate of the conditional
101+
LF probabilities, `P(\lf | Y)`, which are then set as the parameters of the label
102+
model used to re-weight and combine the labels output by the LFs.
103+
104+
Currently this class uses a conditionally independent label model, in which the LFs
105+
are assumed to be conditionally independent given `Y`.
91106
92107
Examples
93108
--------

0 commit comments

Comments
 (0)