From 389a56e0d4a00f235d00c96e4d0e98312f2d6cb6 Mon Sep 17 00:00:00 2001 From: Samir Nasibli Date: Tue, 27 Sep 2022 10:39:22 -0500 Subject: [PATCH] XPU cases for sklearnex DF --- configs/skl_df_xpu_config.json | 34 ++++++++++++++++++++++++++++++++++ 1 file changed, 34 insertions(+) create mode 100644 configs/skl_df_xpu_config.json diff --git a/configs/skl_df_xpu_config.json b/configs/skl_df_xpu_config.json new file mode 100644 index 000000000..1c29ad846 --- /dev/null +++ b/configs/skl_df_xpu_config.json @@ -0,0 +1,34 @@ +{ + "common": { + "lib": "sklearn", + "data-format": "pandas", + "data-order": "F", + "dtype": "float64", + "device": ["host", "cpu", "gpu", "none"] + }, + "cases": [ + { + "algorithm": "df_clsf", + "dtype": "float32", + "max-features": "sqrt", + "dataset": [ + { + "source": "npy", + "name": "susy", + "training": + { + "x": "data/susy_x_train.npy", + "y": "data/susy_y_train.npy" + }, + "testing": + { + "x": "data/susy_x_test.npy", + "y": "data/susy_y_test.npy" + } + } + ], + "num-trees": 100, + "max-depth": 8 + } + ] +}