experiment setup update v1

kumo-ai · Sep 29, 2024 · 0269b3f · 0269b3f
1 parent 9ae68ec
commit 0269b3f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/examples/static_example.py b/examples/static_example.py
@@ -48,7 +48,7 @@
     choices=["hybridgnn", "idgnn", "shallowrhsgnn"],
 )
 parser.add_argument("--lr", type=float, default=0.001)
-parser.add_argument("--epochs", type=int, default=20)
+parser.add_argument("--epochs", type=int, default=1)
 parser.add_argument("--eval_epochs_interval", type=int, default=1)
 parser.add_argument("--batch_size", type=int, default=128)
 parser.add_argument("--supervision_ratio", type=float, default=0.5)
@@ -75,7 +75,7 @@
 TRANSACTION_TABLE = "transaction_table"
 SRC_ENTITY_COL = "user_id"
 DST_ENTITY_COL = "item_id"
-EVAL_K = 12
+EVAL_K = 20
 METRICS = [
     link_prediction_map,
     link_prediction_precision,
@@ -534,6 +534,7 @@ def evaluate(
 
 for epoch in range(1, args.epochs + 1):
     train_loss = train()
+    print(f"{epoch=}, {train_loss=}")
 
 test_pred = test(loader_dict["test"], desc="Test")
 test_metrics = evaluate(test_pred, test_df)