dariolopez
commited on
Commit
·
41959d9
1
Parent(s):
1aa8f09
Update README.md
Browse files
README.md
CHANGED
@@ -14,7 +14,41 @@ The trained model is a fine-tuned version of [PlanTL-GOB-ES/roberta-base-bne](ht
|
|
14 |
|
15 |
# How to use
|
16 |
|
17 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
|
19 |
# Training
|
20 |
|
@@ -41,4 +75,10 @@ In progress
|
|
41 |
"seed": 42,
|
42 |
"length_embedding": 768
|
43 |
}
|
44 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
14 |
|
15 |
# How to use
|
16 |
|
17 |
+
Using this model becomes easy when you have sentence-transformers installed:
|
18 |
+
|
19 |
+
```
|
20 |
+
pip install -U sentence-transformers
|
21 |
+
```
|
22 |
+
|
23 |
+
Then you can use the model like this:
|
24 |
+
|
25 |
+
```python
|
26 |
+
from sentence_transformers import SentenceTransformer, util
|
27 |
+
|
28 |
+
|
29 |
+
# Load model
|
30 |
+
model = SentenceTransformer('dariolopez/roberta-base-bne-finetuned-msmarco-qa-es')
|
31 |
+
|
32 |
+
# Some examples that may contain information that is relevant to your question
|
33 |
+
corpus = [
|
34 |
+
"Napoleón I Bonaparte (Ajaccio, 15 de agosto de 1769-Longwood, 5 de mayo de 1821) fue un militar y estadista francés, general republicano durante la Revolución francesa y el Directorio, y artífice del golpe de Estado del 18 de brumario que lo convirtió en primer cónsul (Premier Consul) de la República el 11 de noviembre de 1799.",
|
35 |
+
"Luis XVI de Francia (en francés: Louis XVI; Versalles, 23 de agosto de 1754 – París, 21 de enero de 1793) fue rey de Francia y de Navarra4 entre 1774 y 1789, copríncipe de Andorra entre 1774 y 1793, y rey de los franceses3 entre 1789 y 1792.2 Fue el último monarca antes de la caída de la monarquía por la Revolución Francesa, así como el último que ejerció sus poderes de monarca absoluto.",
|
36 |
+
"Felipe VI de España (Madrid, 30 de enero de 1968) es el actual rey de España, título por el que ostenta la jefatura del Estado y el mando supremo de las Fuerzas Armadas, desde el 19 de junio de 2014, fecha en que ascendió al trono por la abdicación de su padre, el rey Juan Carlos I.",
|
37 |
+
"Lionel Andrés Messi Cuccittini (Rosario, 24 de junio de 1987), conocido como Leo Messi, es un futbolista argentino que juega como delantero o centrocampista. Jugador histórico del Fútbol Club Barcelona, al que estuvo ligado veinte años, desde 2021 integra el plantel del Paris Saint-Germain de la Ligue 1 de Francia. Es también internacional con la selección de Argentina, equipo del que es capitán."
|
38 |
+
]
|
39 |
+
|
40 |
+
# Your question
|
41 |
+
query = "Listar aquellos personajes que tuvieron poder en Francia"
|
42 |
+
|
43 |
+
# Encode corpus and query
|
44 |
+
corpus_embeddings = model.encode(corpus)
|
45 |
+
query_embedding = model.encode(query)
|
46 |
+
|
47 |
+
# Get the 2 best results on the corpus options
|
48 |
+
hits = util.semantic_search(query_embedding, corpus_embeddings, top_k=2)[0]
|
49 |
+
for hit in hits:
|
50 |
+
print(f"corpus_id: {hit['corpus_id']}, score: {hit['score']}, text: {corpus[hit['corpus_id']][0:100]}...")
|
51 |
+
```
|
52 |
|
53 |
# Training
|
54 |
|
|
|
75 |
"seed": 42,
|
76 |
"length_embedding": 768
|
77 |
}
|
78 |
+
```
|
79 |
+
|
80 |
+
# Additional Information
|
81 |
+
|
82 |
+
## Licesing
|
83 |
+
|
84 |
+
This work is licensed under a [Apache License, Version 2.0](https://www.apache.org/licenses/LICENSE-2.0)
|