class_docstr: Wrapper of FeatureExtractor with the linear classification head.
  
  This class implements the linear classification layer on top of the trained
  mapping from the obfuscated embedddings to the clean ones. This class uses a
  FeatureExtractor model as a part of its pipeline. No assumption is made on the
  trainability of the FeatureExtractor model internally.
  
  Attributes:
    feature_extractor: FeatureExtractor object, which extracts embeddings from
      the obfuscated images and maps them to embeddings of clean images.
    clf_layer: Linear classifier on top of the trained mapping from obfuscated
      to clean embedddings.
class_docstr_tok: ["Wrapper","of","FeatureExtractor","with","the","linear","classification","head",".","This","class","implements","the","linear","classification","layer","on","top","of","the","trained","mapping","from","the","obfuscated","embedddings","to","the","clean","ones",".","This","class","uses","a","FeatureExtractor","model","as","a","part","of","its","pipeline",".","No","assumption","is","made","on","the","trainability","of","the","FeatureExtractor","model","internally",".","Attributes",":","feature_extractor",":","FeatureExtractor","object",",","which","extracts","embeddings","from","the","obfuscated","images","and","maps","them","to","embeddings","of","clean","images",".","clf_layer",":","Linear","classifier","on","top","of","the","trained","mapping","from","obfuscated","to","clean","embedddings","."]
code: def __init__(
      self,
      num_classes: int,
      feature_extractor: FeatureExtractor,
      weight_decay: float = 1e-4,
  ):
      super().__init__()
      self.feature_extractor = feature_extractor
  
      self.clf_layer = tf.keras.layers.Dense(
          num_classes,
          activation=None,
          kernel_regularizer=tf.keras.regularizers.l2(weight_decay),
          bias_regularizer=tf.keras.regularizers.l2(weight_decay),
          name="classifier",
      )
  
imports: import abc
  from typing import Union, Tuple, Sequence, Optional
  import tensorflow
  import tensorflow_hub
label_desc: Use image node_id 1 to create a new FeatureExtractorWithClassifier object from inherited base classes: tf.keras with example: obj = FeatureExtractorWithClassifier(num_classes, feature_extractor, weight_decay)