Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -5,8 +5,6 @@ from fastai.learner import load_learner
|
|
5 |
import pandas as pd
|
6 |
import numpy as np
|
7 |
|
8 |
-
learn = load_learner('model.pkl')
|
9 |
-
|
10 |
dados = pd.read_csv('valid.csv')
|
11 |
ids = dados['user'].unique()
|
12 |
ids_list = list(map(str, ids.tolist()))
|
@@ -43,6 +41,8 @@ class DotProduct(Module):
|
|
43 |
res += self.user_bias(x[:,0]) + self.books_bias(x[:,1])
|
44 |
return sigmoid_range(res, *self.y_range)
|
45 |
|
|
|
|
|
46 |
def top5(user):
|
47 |
user = int(user)
|
48 |
items = pd.Series(learn.dls.classes['title']).unique()
|
|
|
5 |
import pandas as pd
|
6 |
import numpy as np
|
7 |
|
|
|
|
|
8 |
dados = pd.read_csv('valid.csv')
|
9 |
ids = dados['user'].unique()
|
10 |
ids_list = list(map(str, ids.tolist()))
|
|
|
41 |
res += self.user_bias(x[:,0]) + self.books_bias(x[:,1])
|
42 |
return sigmoid_range(res, *self.y_range)
|
43 |
|
44 |
+
learn = load_learner('model.pkl', cpu=True)
|
45 |
+
|
46 |
def top5(user):
|
47 |
user = int(user)
|
48 |
items = pd.Series(learn.dls.classes['title']).unique()
|