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 |
|