test for num bucket

awslabs · Oct 5, 2023 · 355b75b · 355b75b
1 parent 8163a08
commit 355b75b
Showing 1 changed file with 5 additions and 19 deletions.
diff --git a/graphstorm-processing/tests/resources/small_heterogeneous_graph/gconstruct-config.json b/graphstorm-processing/tests/resources/small_heterogeneous_graph/gconstruct-config.json
@@ -19,16 +19,6 @@
         "source_id_col" : "~from",
         "dest_id_col" : "~to",
         "relation": ["user", "rated", "movie"]
-      },
-      {
-        "files": "edges/director-directed-movie.csv",
-        "format": {
-            "name": "csv",
-            "separator" : ","
-        },
-        "source_id_col" : "~from",
-        "dest_id_col" : "~to",
-        "relation": ["director", "directed", "movie"]
       }
     ],
     "nodes" : [
@@ -40,7 +30,7 @@
       },
       {
         "node_id_col":  "~id",
-        "node_type":    "movies",
+        "node_type":    "movie",
         "format":       {"name": "csv", "separator": ","},
         "files":        ["nodes/movie.csv"]
       },
@@ -49,16 +39,12 @@
         "node_type":    "user",
         "format":       {"name": "csv", "separator": ","},
         "files":        ["nodes/user.csv"],
-        "labels" : [
-            {
-              "label_col": "gender",
-              "task_type": "classification",
-              "split_pct" : [0.8, 0.1, 0.1]
-            }
-        ],
         "features": [
             {
-                "feature_col": ["age"]
+                "feature_col": "age",
+                "transform":    {"name": "max_min_norm",
+                          "max_bound": 40,
+                          "min_bound": 4}
             }
         ]
       }