Spaces:
Sleeping
Sleeping
GMARTINEZMILLA
commited on
Commit
•
ef8d112
1
Parent(s):
d837b5d
feat: updated location of files
Browse files
utils.py
CHANGED
@@ -12,14 +12,14 @@ from joblib import dump, load
|
|
12 |
|
13 |
def recomienda_tfid(new_basket):
|
14 |
|
15 |
-
cestas = pd.read_csv('
|
16 |
-
productos = pd.read_csv('
|
17 |
|
18 |
# Cargar la matriz TF-IDF y el modelo
|
19 |
-
tfidf_matrix = load('
|
20 |
# MAtriz que tienen cada columna los diferentes artículos y las diferentes cestas en las filas
|
21 |
# Los valores son la importancia de cada artículo en la cesta según las veces que aparece en la misma y el total de artículos
|
22 |
-
tfidf = load('
|
23 |
|
24 |
# Convertir la nueva cesta en formato TF-IDF
|
25 |
new_basket_str = ' '.join(new_basket)
|
|
|
12 |
|
13 |
def recomienda_tfid(new_basket):
|
14 |
|
15 |
+
cestas = pd.read_csv('cestas.csv')
|
16 |
+
productos = pd.read_csv('productos.csv')
|
17 |
|
18 |
# Cargar la matriz TF-IDF y el modelo
|
19 |
+
tfidf_matrix = load('tfidf_matrix.joblib')
|
20 |
# MAtriz que tienen cada columna los diferentes artículos y las diferentes cestas en las filas
|
21 |
# Los valores son la importancia de cada artículo en la cesta según las veces que aparece en la misma y el total de artículos
|
22 |
+
tfidf = load('tfidf_model.joblib')
|
23 |
|
24 |
# Convertir la nueva cesta en formato TF-IDF
|
25 |
new_basket_str = ' '.join(new_basket)
|