Update app.py
Browse files
app.py
CHANGED
|
@@ -42,8 +42,10 @@ def filtrar_alojamientos(pregunta):
|
|
| 42 |
alojamiento_embedding = tinybert_pipeline(alojamiento)[0][0]
|
| 43 |
|
| 44 |
# Calcular la similitud entre la pregunta y el alojamiento (similitud de coseno)
|
| 45 |
-
|
| 46 |
-
|
|
|
|
|
|
|
| 47 |
|
| 48 |
# Si la similitud es mayor a un umbral, agregar el alojamiento a los resultados
|
| 49 |
if similitud > 0.7: # Umbral de similitud ajustable
|
|
|
|
| 42 |
alojamiento_embedding = tinybert_pipeline(alojamiento)[0][0]
|
| 43 |
|
| 44 |
# Calcular la similitud entre la pregunta y el alojamiento (similitud de coseno)
|
| 45 |
+
producto_punto = sum(p * a for p, a in zip(pregunta_embedding, alojamiento_embedding))
|
| 46 |
+
magnitud_pregunta = (sum(p**2 for p in pregunta_embedding)) ** 0.5
|
| 47 |
+
magnitud_alojamiento = (sum(a**2 for a in alojamiento_embedding)) ** 0.5
|
| 48 |
+
similitud = producto_punto / (magnitud_pregunta * magnitud_alojamiento)
|
| 49 |
|
| 50 |
# Si la similitud es mayor a un umbral, agregar el alojamiento a los resultados
|
| 51 |
if similitud > 0.7: # Umbral de similitud ajustable
|