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