Output Layer

Summary:
Enabling Multi Label output layer.

This is done by migrating domain classifier into a multi label.

Reviewed By: shreydesai

Differential Revision: D25440015

fbshipit-source-id: 3aa8afce8178ec05e8b15617afdeb1b76c50811d

pytext/models/output_layers/doc_classification_output_layer.py

@@ -17,6 +17,7 @@
     KLDivergenceCELoss,
     LabelSmoothedCrossEntropyLoss,
     MultiLabelSoftMarginLoss,
+    BinaryCrossEntropyWithLogitsLoss,
 )
 from pytext.utils.label import get_label_weights
 from torch import jit
@@ -43,6 +44,7 @@ class Config(OutputLayerBase.Config):
         loss: Union[
             CrossEntropyLoss.Config,
             BinaryCrossEntropyLoss.Config,
+            BinaryCrossEntropyWithLogitsLoss.Config,
             MultiLabelSoftMarginLoss.Config,
             AUCPRHingeLoss.Config,
             KLDivergenceBCELoss.Config,
@@ -83,6 +85,8 @@ def from_config(
             cls = BinaryClassificationOutputLayer
         elif isinstance(loss, MultiLabelSoftMarginLoss):
             cls = MultiLabelOutputLayer
+        elif isinstance(loss, BinaryCrossEntropyWithLogitsLoss):
+            cls = MultiLabelOutputLayer
         else:
             cls = MulticlassOutputLayer