Remove references to outputs in contrastive model.

tensorflow / similarity

TensorFlow Similarity is a python package focused on making similarity learning quick and easy.

Apache License 2.0

1.01k stars 104 forks source link

Remove references to outputs in contrastive model. #311

Closed owenvallis closed 1 year ago

owenvallis commented 1 year ago

We use the inputs and outputs to support saving the contrastive model using the Keras API, however, we override train and test steps as well as predict. This means we don't currently support multiple output heads on the embedding output. This PR removes all references to multi-headed outputs and explicitly sets the indexer to use the predictor output.

review-notebook-app[bot] commented 1 year ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.