prhegde commited on
Commit
86fc7ae
1 Parent(s): edc2b91

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -8
README.md CHANGED
@@ -1,15 +1,19 @@
1
- ---
2
- library_name: transformers
3
- tags: []
4
- inference: false
5
- ---
6
 
7
  # Model Card for Model ID
8
  Provides a relevance score for a given search query and a product (with title and description).
9
 
10
  Can be used as retriever/ranker for e-commerce search.
11
 
12
- Development details coming soon.
 
 
 
 
13
 
14
  ## How to use the model
15
  ```python
@@ -37,5 +41,4 @@ output = model(input_ids, attention_mask = attention_masks,
37
  token_type_ids = token_type_ids)
38
  probs = F.softmax(output.logits, dim=-1)
39
  score = probs[0][1].item()
40
- print(score)
41
-
 
1
+ ---
2
+ library_name: transformers
3
+ inference: false
4
+ license: apache-2.0
5
+ ---
6
 
7
  # Model Card for Model ID
8
  Provides a relevance score for a given search query and a product (with title and description).
9
 
10
  Can be used as retriever/ranker for e-commerce search.
11
 
12
+ ## Development details
13
+ Model is trained with a novel adversarial Generator-Retriever framework.
14
+
15
+ The details of the framework can be found [here](https://github.com/PraveenSH/adversarial-generator-retriever/blob/main/README.md).
16
+ Notebook with the code is available [here](https://github.com/PraveenSH/adversarial-generator-retriever/blob/main/generator_retriever.ipynb)
17
 
18
  ## How to use the model
19
  ```python
 
41
  token_type_ids = token_type_ids)
42
  probs = F.softmax(output.logits, dim=-1)
43
  score = probs[0][1].item()
44
+ print(score)