def build_output_model(
self, layers: List[tf.keras.Model]) -> List[tf.keras.Model]:
layers.append(ComputeClassVector(self._input_name, 'cls_vector'))
layers.append(
GlobalVectorPredictor(self._n_classes, 'cls_vector',
self._output_name))
return layers
def build_output_model(
self, layers: List[tf.keras.Model]) -> List[tf.keras.Model]:
if not self.use_global:
layers.append(ComputeClassVector(self._input_name, 'cls_vector'))
else:
layers.append(GlobalExtractor(self._input_name, 'cls_vector'))
layers.append(GlobalVectorPredictor(1, 'cls_vector',
self._output_name))
return layers