Spaces:
Build error
Build error
edit model
Browse files
app.py
CHANGED
|
@@ -3,12 +3,13 @@ import pandas as pd
|
|
| 3 |
import joblib
|
| 4 |
|
| 5 |
data = pd.read_csv(r"data_final.csv")
|
|
|
|
| 6 |
|
| 7 |
def product_recommender(customer_id):
|
| 8 |
list_predicted = []
|
| 9 |
|
| 10 |
for id in data['product_id'].unique():
|
| 11 |
-
preds = list(
|
| 12 |
product_id = preds[1]
|
| 13 |
product_score = preds[3]
|
| 14 |
|
|
|
|
| 3 |
import joblib
|
| 4 |
|
| 5 |
data = pd.read_csv(r"data_final.csv")
|
| 6 |
+
model = joblib.load("KNN_Model.joblib")
|
| 7 |
|
| 8 |
def product_recommender(customer_id):
|
| 9 |
list_predicted = []
|
| 10 |
|
| 11 |
for id in data['product_id'].unique():
|
| 12 |
+
preds = list(model.predict(customer_id, id))
|
| 13 |
product_id = preds[1]
|
| 14 |
product_score = preds[3]
|
| 15 |
|