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