Sheshera Mysore commited on
Commit
586b6c4
1 Parent(s): 8dcc1b3
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -43,10 +43,12 @@ The `aspire-contextualsentence-multi-compsci` model can be used via the `transfo
43
 
44
  Since the model relies on contextual sentence embeddings additional code is needed to obtain cthese from a transformer model. Use it per this example usage script: [`aspire/examples/ex_aspire_consent.py`](https://github.com/allenai/aspire/blob/main/examples/ex_aspire_consent.py)
45
 
46
- Further, since the model relies on computing a document-document scores via a Wasserstein distance, the linked class for computing these distances: [`AllPairMaskedWasserstein`](https://github.com/allenai/aspire/blob/07fdfd08698baa9e17601cc541ac9929694613b6/src/learning/facetid_models/pair_distances.py#L14)
47
 
48
  This code uses the [`geomloss`](https://www.kernel-operations.io/geomloss/api/pytorch-api.html) library for computing Wasserstein distances.
49
 
 
 
50
  ### Variable and metrics
51
  This model is evaluated on information retrieval datasets with document level queries. Performance here is reported on CSFCube (computer science/English). This is detailed on [github](https://github.com/allenai/aspire) and in our [paper](https://arxiv.org/abs/2111.08366). CSFCube presents a finer-grained query via selected sentences in a query abstract based on which a finer-grained retrieval must be made from candidate abstracts.
52
 
 
43
 
44
  Since the model relies on contextual sentence embeddings additional code is needed to obtain cthese from a transformer model. Use it per this example usage script: [`aspire/examples/ex_aspire_consent.py`](https://github.com/allenai/aspire/blob/main/examples/ex_aspire_consent.py)
45
 
46
+ Further, since the model relies on computing a document-document scores via a Wasserstein distance, use the linked class for computing these distances: [`AllPairMaskedWasserstein`](https://github.com/allenai/aspire/blob/07fdfd08698baa9e17601cc541ac9929694613b6/src/learning/facetid_models/pair_distances.py#L14)
47
 
48
  This code uses the [`geomloss`](https://www.kernel-operations.io/geomloss/api/pytorch-api.html) library for computing Wasserstein distances.
49
 
50
+ (Additional example code to demo this more will be added in the next couple days!)
51
+
52
  ### Variable and metrics
53
  This model is evaluated on information retrieval datasets with document level queries. Performance here is reported on CSFCube (computer science/English). This is detailed on [github](https://github.com/allenai/aspire) and in our [paper](https://arxiv.org/abs/2111.08366). CSFCube presents a finer-grained query via selected sentences in a query abstract based on which a finer-grained retrieval must be made from candidate abstracts.
54