added kin8mn preprocess

testbed/src/testbed/data/links.json

@@ -13,5 +13,5 @@
     "news": "https://archive.ics.uci.edu/static/public/332/online+news+popularity.zip",
     "power": "https://archive.ics.uci.edu/static/public/294/combined+cycle+power+plant.zip",
     "superconductor": "https://archive.ics.uci.edu/static/public/464/superconductivty+data.zip",
-    "wave": "https://archive.ics.uci.edu/static/public/494/wave+energy+converters.zip"
+    "wave": "https://archive.ics.uci.edu/static/public/494/wave+energy+converters.zip",
 }

testbed/src/testbed/data/uci/kin8nm/preprocess.py

@@ -0,0 +1,27 @@
+import argparse
+from pathlib import Path
+import numpy as np
+import zipfile
+import pandas as pd
+
+def main(path_raw_dataset_dir: Path):
+    # Process the raw data file named temp.csv
+    raw_data_path = path_raw_dataset_dir / "data.zip"
+    # unzip the data
+    with zipfile.ZipFile(raw_data_path, 'r') as zip_ref:
+        zip_ref.extractall(path_raw_dataset_dir)
+    raw_data_path = path_raw_dataset_dir / "data.csv"
+
+    df = pd.read_csv(raw_data_path)
+    y = df["y"].values
+    x = df.drop(columns=["y"]).values
+    categorical = []
+
+    # Save preprocessed data
+    np.save(path_raw_dataset_dir.parent / "data.npy", {"x": x, "y": y, "categorical": categorical})
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument("path", type=Path)
+    args = parser.parse_args()
+    main(args.path)