support time window

dmlc · Jun 11, 2024 · 228189c · 228189c
1 parent d281e42
commit 228189c
Show file tree

Hide file tree

Showing 5 changed files with 624 additions and 620 deletions.
diff --git a/examples/sampling/graphbolt/temporal_link_prediction.py b/examples/sampling/graphbolt/temporal_link_prediction.py
@@ -30,6 +30,7 @@
 │
 └───> Test set evaluation
 """
+
 import argparse
 import os
 import time
@@ -282,7 +283,7 @@ def main(args):
     print("Loading data")
     # TODO: Add the datasets to built-in.
     dataset_path = download_datasets(args.dataset)
-    dataset = gb.OnDiskDataset(dataset_path).load()
+    dataset = gb.OnDiskDataset(dataset_path, force_preprocess=True).load()
 
     # Move the dataset to the selected storage.
     graph = dataset.graph.to(args.storage_device)