Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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/
|
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/
|
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):
|