google-research / t5x_retrieval

Apache License 2.0
95 stars 9 forks source link

Semantic Textual Similarity format? #1

Open Chroner2 opened 2 years ago

Chroner2 commented 2 years ago

Is it possible to finetune with a custom dataset on an STS format? In specific, have "sentence #1", "sentence #2" and their similarity score (0-5). The provided examples use only positive retrievals as an example. On the tasks.py the key mapping is only inputs and targets.

Many thanks