Javier Real commited on
Commit
ab1e8ef
·
1 Parent(s): 73969b9

Similarity type selector

Browse files
Files changed (1) hide show
  1. app.py +10 -3
app.py CHANGED
@@ -35,8 +35,15 @@ suggestions = {
35
  # Widget para seleccionar el número de resultados
36
  n_results = st.number_input('Number of recommendations', min_value=1, max_value=100, value=10)
37
 
 
 
 
 
 
 
 
 
 
38
  if st.button('Get Recommendations'):
39
- # st.write("Debugging Suggestions:", suggestions)
40
- recommendations = get_recommendations_text(suggestions, n_results=n_results, similarity=EUCLIDEAN_SIMILARITY)
41
- # st.write("Debugging Recommendations:\n", recommendations)
42
  st.markdown(recommendations.replace("\n", "\n\n"))
 
35
  # Widget para seleccionar el número de resultados
36
  n_results = st.number_input('Number of recommendations', min_value=1, max_value=100, value=10)
37
 
38
+ # Widget para seleccionar el método de similitud
39
+ similarity_method = st.radio("Select Similarity Method", ["Cosine Similarity", "Euclidean Similarity"])
40
+
41
+ # Asignar el valor de la constante apropiada basado en la selección del usuario
42
+ if similarity_method == "Cosine Similarity":
43
+ similarity = COSINE_SIMILARITY
44
+ else:
45
+ similarity = EUCLIDEAN_SIMILARITY
46
+
47
  if st.button('Get Recommendations'):
48
+ recommendations = get_recommendations_text(suggestions, n_results=n_results, similarity=similarity)
 
 
49
  st.markdown(recommendations.replace("\n", "\n\n"))