Add __str__ to item-item

lenskit · Sep 13, 2018 · 6e54267 · 6e54267
1 parent d1aba29
commit 6e54267
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lenskit/algorithms/item_knn.py b/lenskit/algorithms/item_knn.py
@@ -239,3 +239,6 @@ def load_model(self, file):
             matrix = sps.csr_matrix((values, indices, indptr))
 
             return IIModel(items, means, np.diff(indptr), matrix, ratings)
+
+    def __str__(self):
+        return 'ItemItem(nnbrs={}, msize={})'.format(self.max_neighbors, self.save_neighbors)