def build_output_model(
self, layers: List[tf.keras.Model]) -> List[tf.keras.Model]:
layers.append(
BidirectionalOutputShift(self._input_name, 'shifted_logits'))
layers.append(
AminoAcidClassPredictor(self._n_classes,
'shifted_logits',
self._output_name,
use_conv=False))
return layers
def build_output_model(
self, layers: List[tf.keras.Model]) -> List[tf.keras.Model]:
layers.insert(
0,
RandomSequenceMask(self._n_classes, self._mask_token,
self._percentage, self._style))
layers.append(
AminoAcidClassPredictor(self._n_classes,
self._input_name,
self._output_name,
use_conv=True))
return layers
def build_output_model(
self, layers: List[tf.keras.Model]) -> List[tf.keras.Model]:
layers.append(
AminoAcidClassPredictor(self._n_classes, self._input_name,
self._output_name))
return layers