Improved examples [skip ci]

ankane · ankane · commit 16e7474a9209 · 2023-03-31T23:28:05.000-07:00
diff --git a/examples/implicit_recs.py b/examples/implicit_recs.py
@@ -34,8 +34,8 @@ class Item(Base):
 model = implicit.als.AlternatingLeastSquares(factors=20)
 model.fit(ratings)
 
-users = [dict(id=i, factors=factors) for i, factors in enumerate(model.user_factors)]
-items = [dict(id=i, title=titles[i], factors=factors) for i, factors in enumerate(model.item_factors)]
+users = [dict(factors=factors) for i, factors in enumerate(model.user_factors)]
+items = [dict(title=titles[i], factors=factors) for i, factors in enumerate(model.item_factors)]
 
 session = Session(engine)
 session.bulk_insert_mappings(User, users)
diff --git a/examples/lightfm_recs.py b/examples/lightfm_recs.py
@@ -38,8 +38,8 @@ class Item(Base):
 user_biases, user_factors = model.get_user_representations()
 item_biases, item_factors = model.get_item_representations()
 
-users = [dict(id=i, factors=factors) for i, factors in enumerate(user_factors)]
-items = [dict(id=i, title=data['item_labels'][i], factors=factors, bias=item_biases[i].item()) for i, factors in enumerate(item_factors)]
+users = [dict(factors=factors) for i, factors in enumerate(user_factors)]
+items = [dict(title=data['item_labels'][i], factors=factors, bias=item_biases[i].item()) for i, factors in enumerate(item_factors)]
 
 session = Session(engine)
 session.bulk_insert_mappings(User, users)
