DanielRN commited on
Commit
3628b32
·
verified ·
1 Parent(s): 9532736

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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
- similitud = sum(p * a for p, a in zip(pregunta_embedding, alojamiento_embedding)) / (
46
- (sum(p**2 for p in pregunta_embedding) ** 0.5) * (sum(a**2 for a in alojamiento_embedding) ** 0.5)
 
 
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