mariaorri commited on
Commit
486c75e
·
verified ·
1 Parent(s): 1c56862

Update utils.py

Browse files

Cambio en las direcciones de los csv y joblib

Files changed (1) hide show
  1. utils.py +3 -3
utils.py CHANGED
@@ -9,8 +9,8 @@ from sklearn.preprocessing import normalize
9
 
10
  def recomienda_tf(new_basket, cestas, productos):
11
  # Cargar la matriz TF y el modelo
12
- tf_matrix = load('tf_matrix.joblib')
13
- count = load('count_vectorizer.joblib')
14
 
15
  # Convertir la nueva cesta en formato TF (Term Frequency)
16
  new_basket_str = ' '.join(new_basket)
@@ -68,7 +68,7 @@ def retroalimentacion(cestas, cesta_nueva):
68
  cestas.loc[len(cestas)] = cesta_unida
69
  print("Cesta añadida.")
70
  # Reescribimos la nueva cesta
71
- cestas.to_csv('cesta_su.csv')
72
  else:
73
  print("La cesta ya existe en el DataFrame.")
74
 
 
9
 
10
  def recomienda_tf(new_basket, cestas, productos):
11
  # Cargar la matriz TF y el modelo
12
+ tf_matrix = load('count_matrix_2.joblib')
13
+ count = load('count_vectorizer_2.joblib')
14
 
15
  # Convertir la nueva cesta en formato TF (Term Frequency)
16
  new_basket_str = ' '.join(new_basket)
 
68
  cestas.loc[len(cestas)] = cesta_unida
69
  print("Cesta añadida.")
70
  # Reescribimos la nueva cesta
71
+ cestas.to_csv('cestas_final.csv')
72
  else:
73
  print("La cesta ya existe en el DataFrame.")
74