src/transformers/models/esm/modeling_tf_esm.py/TFEsmContactPredictionHead/__init__
class TFEsmContactPredictionHead:    def __init__(
        self,
        in_features: int,
        bias=True,
        eos_idx: int = 2,
        name=None,
    ):
        super().__init__(name=name)
        self.eos_idx = eos_idx
        self.in_features = in_features
        self.regression = Dense(1, use_bias=bias, activation="sigmoid", name="regression")