add __str__ to user-user

lenskit · Sep 13, 2018 · 2dc0e97 · 2dc0e97
1 parent 6e54267
commit 2dc0e97
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lenskit/algorithms/user_knn.py b/lenskit/algorithms/user_knn.py
@@ -190,3 +190,6 @@ def save_model(self, model, file):
     def load_model(self, file):
         with pd.HDFStore(file, 'r') as store:
             return UUModel(store['matrix'], store['stats'], store['item_users'])
+
+    def __str__(self):
+        return 'UserUser(nnbrs={}, min_sim={})'.format(self.max_neighbors, self.min_similarity)