csclarke commited on
Commit
ed62b56
·
1 Parent(s): 95b6f74

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +22 -22
README.md CHANGED
@@ -1,22 +1,22 @@
1
- ---
2
- license: cc
3
- ---
4
- # MARS Encoder for Multi-agent Response Selection
5
- This model was trained using [SentenceTransformers](https://sbert.net) [Cross-Encoder](https://www.sbert.net/examples/applications/cross-encoder/README.html) class and is the model used in the paper [One Agent To Rule Them All: Towards Multi-agent Conversational AI](https://csclarke.com/assets/pdf/ACL_2022.pdf).
6
-
7
- ## Training Data
8
- This model was trained on the [BBAI dataset](https://github.com/ChrisIsKing/black-box-multi-agent-integation/tree/main/data). The model will predict a score between 0 and 1 ranking the correctness of a response to a user question from a conversational agent.
9
-
10
-
11
- ## Usage and Performance
12
-
13
- Pre-trained models can be used like this:
14
- ```python
15
- from sentence_transformers import CrossEncoder
16
- model = CrossEncoder('csclarke/MARS-Encoder')
17
- scores = model.predict([('question 1', 'response 1'), ('question 1', 'response 2')])
18
- ```
19
-
20
- The model will predict scores for the pairs `('question 1', 'response 1')` and `('question 1', 'response 2')`.
21
-
22
- You can use this model also without sentence_transformers and by just using Transformers ``AutoModel`` class
 
1
+ ---
2
+ license: cc
3
+ ---
4
+ # MARS Encoder for Multi-Agent Response Selection
5
+ This model was trained using [SentenceTransformers](https://sbert.net) [Cross-Encoder](https://www.sbert.net/examples/applications/cross-encoder/README.html) class and is the model used in the paper [One Agent To Rule Them All: Towards Multi-agent Conversational AI](https://csclarke.com/assets/pdf/ACL_2022.pdf).
6
+
7
+ ## Training Data
8
+ This model was trained on the [BBAI dataset](https://github.com/ChrisIsKing/black-box-multi-agent-integation/tree/main/data). The model will predict a score between 0 and 1 ranking the correctness of a response to a user question from a conversational agent.
9
+
10
+
11
+ ## Usage and Performance
12
+
13
+ Pre-trained models can be used like this:
14
+ ```python
15
+ from sentence_transformers import CrossEncoder
16
+ model = CrossEncoder('csclarke/MARS-Encoder')
17
+ scores = model.predict([('question 1', 'response 1'), ('question 1', 'response 2')])
18
+ ```
19
+
20
+ The model will predict scores for the pairs `('question 1', 'response 1')` and `('question 1', 'response 2')`.
21
+
22
+ You can use this model also without sentence_transformers and by just using Transformers ``AutoModel`` class