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 + } + ] +}