diff --git a/configs/xpu/dbscan.json b/configs/xpu/dbscan.json index b6b0f7938..68ced8668 100644 --- a/configs/xpu/dbscan.json +++ b/configs/xpu/dbscan.json @@ -19,6 +19,7 @@ } } ], + "workload-size": "small", "eps": 5, "min-samples": 3 }, @@ -33,6 +34,7 @@ } } ], + "workload-size": "medium", "eps": 1.7e3, "min-samples": 3 }, @@ -47,6 +49,7 @@ } } ], + "workload-size": "small", "eps": 1.0e3, "min-samples": 220 } diff --git a/configs/xpu/df_regr.json b/configs/xpu/df_regr.json index 7ad6165b0..3c6c05bfd 100644 --- a/configs/xpu/df_regr.json +++ b/configs/xpu/df_regr.json @@ -26,7 +26,8 @@ } } ], - "num-trees": [10, 100], + "workload-size": "medium", + "num-trees": 10, "max-depth": 5 }, { @@ -46,7 +47,50 @@ } } ], - "num-trees": [100, 20], + "workload-size": "large", + "num-trees": 10, + "max-depth": 5 + }, + { + "dataset": [ + { + "source": "npy", + "name": "year_prediction_msd", + "training": + { + "x": "data/year_prediction_msd_x_train.npy", + "y": "data/year_prediction_msd_y_train.npy" + }, + "testing": + { + "x": "data/year_prediction_msd_x_test.npy", + "y": "data/year_prediction_msd_y_test.npy" + } + } + ], + "workload-size": "medium", + "num-trees": 20, + "max-depth": 8 + }, + { + "dataset": [ + { + "source": "npy", + "name": "year_prediction_msd", + "training": + { + "x": "data/year_prediction_msd_x_train.npy", + "y": "data/year_prediction_msd_y_train.npy" + }, + "testing": + { + "x": "data/year_prediction_msd_x_test.npy", + "y": "data/year_prediction_msd_y_test.npy" + } + } + ], + "workload-size": "large", + "num-trees": 100, "max-depth": 8 }, { @@ -66,6 +110,7 @@ } } ], + "workload-size": "large", "num-trees": 20, "max-depth": 16 }, @@ -86,7 +131,29 @@ } } ], - "num-trees": [15, 20, 100], + "workload-size": "medium", + "num-trees": [15, 20], + "max-depth": 8 + }, + { + "dataset": [ + { + "source": "npy", + "name": "higgs1m", + "training": + { + "x": "data/higgs1m_x_train.npy", + "y": "data/higgs1m_y_train.npy" + }, + "testing": + { + "x": "data/higgs1m_x_test.npy", + "y": "data/higgs1m_y_test.npy" + } + } + ], + "workload-size": "large", + "num-trees": 100, "max-depth": 8 }, { @@ -106,6 +173,7 @@ } } ], + "workload-size": "large", "num-trees": 100, "max-depth": 8 }, @@ -126,6 +194,7 @@ } } ], + "workload-size": "large", "num-trees": 20, "max-depth": 16 } diff --git a/configs/xpu/kmeans.json b/configs/xpu/kmeans.json index 1e94d892b..61c191fe3 100644 --- a/configs/xpu/kmeans.json +++ b/configs/xpu/kmeans.json @@ -19,6 +19,7 @@ } } ], + "workload-size": "medium", "n-clusters": 10, "maxiter": 100 }, @@ -33,7 +34,23 @@ } } ], - "n-clusters": [100, 250], + "workload-size": "small", + "n-clusters": 100, + "maxiter": 10 + }, + { + "dataset": [ + { + "source": "npy", + "name": "higgs_one_m_clustering", + "training": + { + "x": "data/higgs_one_m_clustering.npy" + } + } + ], + "workload-size": "medium", + "n-clusters": 250, "maxiter": 10 }, { @@ -47,6 +64,7 @@ } } ], + "workload-size": "medium", "n-clusters": [512, 1024, 2048], "maxiter": 10 }, @@ -61,7 +79,23 @@ } } ], - "n-clusters": [100, 250], + "workload-size": "small", + "n-clusters": 100, + "maxiter": 10 + }, + { + "dataset": [ + { + "source": "npy", + "name": "hepmass_1M_cluster", + "training": + { + "x": "data/hepmass_1M_cluster.npy" + } + } + ], + "workload-size": "medium", + "n-clusters": 250, "maxiter": 10 }, { @@ -75,6 +109,7 @@ } } ], + "workload-size": "medium", "n-clusters": 10, "maxiter": 100 }, @@ -89,6 +124,7 @@ } } ], + "workload-size": "medium", "n-clusters": 10, "maxiter": 100 }, @@ -103,6 +139,7 @@ } } ], + "workload-size": "medium", "n-clusters": [100 , 250], "maxiter": 10 }, @@ -117,6 +154,7 @@ } } ], + "workload-size": "medium", "n-clusters": [512, 1024, 2048], "maxiter": 10 } diff --git a/configs/xpu/knn_clsf.json b/configs/xpu/knn_clsf.json index d13128f2f..0e352f262 100644 --- a/configs/xpu/knn_clsf.json +++ b/configs/xpu/knn_clsf.json @@ -25,6 +25,7 @@ } } ], + "workload-size": "medium", "n-neighbors": [2, 100] }, { @@ -58,6 +59,7 @@ } } ], + "workload-size": "medium", "n-neighbors": [5, 100] }, { @@ -77,6 +79,7 @@ } } ], + "workload-size": "medium", "n-neighbors": 7 }, { @@ -96,6 +99,7 @@ } } ], + "workload-size": "medium", "n-neighbors": 5 }, { @@ -115,6 +119,7 @@ } } ], + "workload-size": "medium", "task": "search", "n-neighbors": 2 }, @@ -135,6 +140,7 @@ } } ], + "workload-size": "medium", "task": "search", "n-neighbors": 5 }, @@ -155,6 +161,7 @@ } } ], + "workload-size": "medium", "task": "search", "n-neighbors": 7 } diff --git a/configs/xpu/knn_regr.json b/configs/xpu/knn_regr.json index 32aa47c3b..0861a855d 100644 --- a/configs/xpu/knn_regr.json +++ b/configs/xpu/knn_regr.json @@ -25,6 +25,7 @@ } } ], + "workload-size": "medium", "n-neighbors": 2 }, { @@ -44,6 +45,7 @@ } } ], + "workload-size": "medium", "n-neighbors": 5 }, { @@ -63,6 +65,7 @@ } } ], + "workload-size": "medium", "n-neighbors": 7 } ] diff --git a/configs/xpu/linear.json b/configs/xpu/linear.json index 93dce36e4..281f97135 100644 --- a/configs/xpu/linear.json +++ b/configs/xpu/linear.json @@ -24,7 +24,8 @@ "y": "data/year_prediction_msd_y_test.npy" } } - ] + ], + "workload-size": "small" }, { "dataset": [ @@ -42,7 +43,8 @@ "y": "data/higgs1m_y_test.npy" } } - ] + ], + "workload-size": "small" } ] } diff --git a/configs/xpu/log_reg.json b/configs/xpu/log_reg.json index bf62fee5d..b65ac0316 100644 --- a/configs/xpu/log_reg.json +++ b/configs/xpu/log_reg.json @@ -25,6 +25,7 @@ } } ], + "workload-size": "medium", "maxiter": "20" }, { @@ -44,6 +45,7 @@ } } ], + "workload-size": "medium", "maxiter": "10" }, { @@ -63,6 +65,7 @@ } } ], + "workload-size": "small", "no-fit-intercept": "", "maxiter": "50" }, @@ -83,6 +86,7 @@ } } ], + "workload-size": "medium", "maxiter": "500" } ] diff --git a/configs/xpu/pca.json b/configs/xpu/pca.json index e13f1acb6..f282bc2c2 100644 --- a/configs/xpu/pca.json +++ b/configs/xpu/pca.json @@ -19,7 +19,8 @@ "y": "data/higgs1m_y_train.npy" } } - ] + ], + "workload-size": "small" }, { "device": ["host", "cpu", "gpu", "none"], @@ -33,7 +34,8 @@ "y": "data/epsilon_30K_y_train.npy" } } - ] + ], + "workload-size": "small" }, { "dataset": [ @@ -46,7 +48,8 @@ "y": "data/cifar_binary_y_train.npy" } } - ] + ], + "workload-size": "medium" } ] } diff --git a/configs/xpu/svm.json b/configs/xpu/svm.json index 667622050..74d7713fe 100644 --- a/configs/xpu/svm.json +++ b/configs/xpu/svm.json @@ -25,6 +25,7 @@ } } ], + "workload-size": "small", "C": 1.5e-3, "kernel": "linear" }, @@ -45,6 +46,7 @@ } } ], + "workload-size": "large", "C": 1.0, "kernel": "linear" }, @@ -65,6 +67,7 @@ } } ], + "workload-size": "large", "C": 1.0, "kernel": "linear" }, @@ -85,6 +88,7 @@ } } ], + "workload-size": "medium", "C": 1.0e-7, "kernel": "linear" }, @@ -105,6 +109,7 @@ } } ], + "workload-size": "medium", "C": 9.0e2, "kernel": "rbf" }, @@ -125,6 +130,7 @@ } } ], + "workload-size": "medium", "C": 1000.0, "kernel": "rbf" }, @@ -145,6 +151,7 @@ } } ], + "workload-size": "medium", "C": 1000.0, "kernel": "rbf" }