Update app.py

app.py

@@ -3,6 +3,21 @@ import joblib
 
 #__all__ = ['NumberClass', 'learn', 'classify_image', 'categories', 'image', 'label', 'examples', 'intf']
 
+class DotProductBias(Module):
+    def __init__(self, n_users, n_animes, n_factors, y_range=(0,10.5)):
+        self.user_factors = Embedding(n_users, n_factors)
+        self.user_bias = Embedding(n_users, 1)
+        self.anime_factors = Embedding(n_animes, n_factors)
+        self.anime_bias = Embedding(n_animes, 1)
+        self.y_range = y_range
+        
+    def forward(self, x):
+        users = self.user_factors(x[:,0])
+        animes = self.anime_factors(x[:,1])
+        res = (users * animes).sum(dim=1, keepdim=True)
+        res += self.user_bias(x[:,0]) + self.anime_bias(x[:,1])
+        return sigmoid_range(res, *self.y_range)
+
 def get_y(r):
     number = ''.join(filter(str.isdigit, str(r)))
     return train_y[int(number)]