code: def _get_classifier(self, prefix):
          
          with self.name_scope():
              classifier = nn.Dense(2, prefix=prefix)
          return classifier
docs: Construct a decoder for the next sentence prediction task