Example #1
0
 class Config(ConfigBase):
     representation: SeqRepresentation.Config = SeqRepresentation.Config()
     output_layer: ClassificationOutputLayer.Config = (
         ClassificationOutputLayer.Config())
     decoder: MLPDecoder.Config = MLPDecoder.Config()
 class Config(BaseModel.Config):
     decoder: MLPDecoder.Config = MLPDecoder.Config()
     output_layer: ClassificationOutputLayer.Config = (
         ClassificationOutputLayer.Config()
     )
     encode_relations: bool = True
 class Config(BaseModel.Config):
     decoder: MLPDecoder.Config = MLPDecoder.Config()
     output_layer: Union[
         ClassificationOutputLayer.Config, PairwiseCosineDistanceOutputLayer.Config
     ] = ClassificationOutputLayer.Config()
     encode_relations: bool = True
 class Config(ConfigBase):
     representation: PairRepresentation.Config = PairRepresentation.Config()
     decoder: MLPDecoder.Config = MLPDecoder.Config()
     # TODO: will need to support different output layer for contrastive loss
     output_layer: ClassificationOutputLayer.Config = (
         ClassificationOutputLayer.Config())
            def __init__(self, in_dim, out_dim, temp):
                super().__init__()

                self.mlp = MLPDecoder.from_config(
                    MLPDecoder.Config(bias=False, temperature=temp), in_dim, out_dim
                )