src/proteingnn/model.py
def _set_final_layer(self):
        """Set final layer depending on regression or classification task."""
        if self._regression:
            return nn.LazyLinear(out_features=1)
        else:
            return nn.LazyLinear(out_features=self.num_classes)
Set final layer depending on regression or classification task.
Set final layer depending on regression or classification task.
Set final layer depending on regression or classification task .
def _set_final_layer(self):
        if self._regression:
            return nn.LazyLinear(out_features=1)
        else:
            return nn.LazyLinear(out_features=self.num_classes)
def _set_final_layer ( self ) : if self . _regression : return nn . LazyLinear ( out_features = 1 ) else : return nn . LazyLinear ( out_features = self . num_classes )
Set final layer depending on regression or classification task.
Set final layer depending on regression or classification task .
"""Set final layer depending on regression or classification task."""
self