shirwu commited on
Commit
c83dcbf
·
1 Parent(s): a00d62c

update link

Browse files
src/benchmarks/semistruct/amazon.py CHANGED
@@ -18,7 +18,7 @@ from src.tools.io import save_files, load_files
18
 
19
  PROCESSED_DATASET = {
20
  "repo": "snap-stanford/STaRK-Dataset",
21
- "file": "amazon_processed.zip",
22
  }
23
 
24
  class AmazonSemiStruct(SemiStructureKB):
 
18
 
19
  PROCESSED_DATASET = {
20
  "repo": "snap-stanford/STaRK-Dataset",
21
+ "file": "skb/amazon_processed.zip",
22
  }
23
 
24
  class AmazonSemiStruct(SemiStructureKB):
src/benchmarks/semistruct/mag.py CHANGED
@@ -17,8 +17,8 @@ from src.benchmarks.semistruct.knowledge_base import SemiStructureKB
17
  from src.tools.io import save_files, load_files
18
 
19
  PROCESSED_DATASET = {
20
- "repo": "snap-stanford/STaRK-Dataset",
21
- "file": "mag_processed.zip",
22
  }
23
 
24
  class MagSemiStruct(SemiStructureKB):
 
17
  from src.tools.io import save_files, load_files
18
 
19
  PROCESSED_DATASET = {
20
+ "repo": "snap-stanford/stark",
21
+ "file": "skb/mag_processed.zip",
22
  }
23
 
24
  class MagSemiStruct(SemiStructureKB):
src/benchmarks/semistruct/primekg.py CHANGED
@@ -14,8 +14,8 @@ import zipfile
14
  import json
15
 
16
  PROCESSED_DATASET = {
17
- "repo": "snap-stanford/STaRK-Dataset",
18
- "file": "primekg_processed.zip",
19
  }
20
 
21
  class PrimeKGSemiStruct(SemiStructureKB):
 
14
  import json
15
 
16
  PROCESSED_DATASET = {
17
+ "repo": "snap-stanford/stark",
18
+ "file": "skb/primekg_processed.zip",
19
  }
20
 
21
  class PrimeKGSemiStruct(SemiStructureKB):