Skip to content

Commit 3f25ddb

Browse files
committedMar 21, 2025
Merge branch 'dev/eglaser-large-scale-ci-fix' into dev/eglaser-large-configs
2 parents fca41a9 + 66b6834 commit 3f25ddb

11 files changed

+91
-392
lines changed
 

‎configs/incremental.json

-100
This file was deleted.
+77
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,77 @@
1+
{ "INCLUDE": [ ],
2+
"PARAMETERS_SETS": {
3+
"common incremental raw gpu params": {
4+
"algorithm": {
5+
"device": "gpu",
6+
"sklearnex_context": { "use_raw_input": true }
7+
},
8+
"data": {
9+
"format":"dpctl",
10+
"order": "C"
11+
}
12+
},
13+
"statistical batches and data": [
14+
{ "algorithm": { "num_batches": { "training": [1,2,6,12]} }, "data": { "source": "make_blobs", "generation_kwargs": { "n_samples": 100000000, "n_features": 10, "centers": 1 } } },
15+
{ "algorithm": { "num_batches": { "training": 2} }, "data": { "source": "make_blobs", "generation_kwargs": { "n_samples": 50000000, "n_features": 10, "centers": 1 } } },
16+
{ "algorithm": { "num_batches": { "training": 6} }, "data": { "source": "make_blobs", "generation_kwargs": { "n_samples": 16666667, "n_features": 10, "centers": 1 } } },
17+
{ "algorithm": { "num_batches": { "training": 12} }, "data": { "source": "make_blobs", "generation_kwargs": { "n_samples": 8333333, "n_features": 10, "centers": 1 } } },
18+
{ "algorithm": { "num_batches": { "training": [1,2,6,12]} }, "data": { "source": "make_blobs", "generation_kwargs": { "n_samples": 1000000, "n_features": 1000, "centers": 1 } } },
19+
{ "algorithm": { "num_batches": { "training": 2} }, "data": { "source": "make_blobs", "generation_kwargs": { "n_samples": 500000, "n_features": 1000, "centers": 1 } } },
20+
{ "algorithm": { "num_batches": { "training": 6} }, "data": { "source": "make_blobs", "generation_kwargs": { "n_samples": 166667, "n_features": 1000, "centers": 1 } } },
21+
{ "algorithm": { "num_batches": { "training": 12} }, "data": { "source": "make_blobs", "generation_kwargs": { "n_samples": 83333, "n_features": 1000, "centers": 1 } } }
22+
],
23+
"regression batches and data": [
24+
{ "algorithm": { "num_batches": { "training": [1,2,6,12]} }, "data": { "source": "make_regression", "generation_kwargs": { "n_samples": 100005000, "n_features": 10, "noise": 1.25 }, "split_kwargs": { "train_size": 100000000, "test_size": 5000 } } },
25+
{ "algorithm": { "num_batches": { "training": 2} }, "data": { "source": "make_regression", "generation_kwargs": { "n_samples": 100005000, "n_features": 10, "noise": 1.25 }, "split_kwargs": { "train_size": 50000000, "test_size": 5000 } } },
26+
{ "algorithm": { "num_batches": { "training": 6} }, "data": { "source": "make_regression", "generation_kwargs": { "n_samples": 100005000, "n_features": 10, "noise": 1.25 }, "split_kwargs": { "train_size": 16666667, "test_size": 5000 } } },
27+
{ "algorithm": { "num_batches": { "training": 12} }, "data": { "source": "make_regression", "generation_kwargs": { "n_samples": 100005000, "n_features": 10, "noise": 1.25 }, "split_kwargs": { "train_size": 8333333, "test_size": 5000 } } },
28+
{ "algorithm": { "num_batches": { "training": [1,2,6,12]} }, "data": { "source": "make_regression", "generation_kwargs": { "n_samples": 3005000, "n_features": 1000, "noise": 1.25 }, "split_kwargs": { "train_size": 3000000, "test_size": 5000 } } },
29+
{ "algorithm": { "num_batches": { "training": 2} }, "data": { "source": "make_regression", "generation_kwargs": { "n_samples": 3005000, "n_features": 1000, "noise": 1.25 }, "split_kwargs": { "train_size": 1500000, "test_size": 5000 } } },
30+
{ "algorithm": { "num_batches": { "training": 6} }, "data": { "source": "make_regression", "generation_kwargs": { "n_samples": 3005000, "n_features": 1000, "noise": 1.25 }, "split_kwargs": { "train_size": 500000, "test_size": 5000 } } },
31+
{ "algorithm": { "num_batches": { "training": 12} }, "data": { "source": "make_regression", "generation_kwargs": { "n_samples": 3005000, "n_features": 1000, "noise": 1.25 }, "split_kwargs": { "train_size": 250000, "test_size": 5000 } } }
32+
],
33+
"covariance": {
34+
"algorithm": {
35+
"estimator": "IncrementalEmpiricalCovariance",
36+
"library": "sklearnex",
37+
"estimator_methods": {"training": "partial_fit"}
38+
},
39+
"data": {
40+
"split_kwargs": { "test_size": 0.0001 }
41+
}
42+
},
43+
"basic_statistics": {
44+
"algorithm": {
45+
"estimator": "IncrementalBasicStatistics",
46+
"library": "sklearnex",
47+
"estimator_methods": {"training": "partial_fit"}
48+
},
49+
"data": {
50+
"split_kwargs": { "test_size": 0.0001 }
51+
}
52+
},
53+
"linear_regression": {
54+
"algorithm": {
55+
"estimator": "IncrementalLinearRegression",
56+
"library": "sklearnex",
57+
"estimator_methods": {"training": "partial_fit"}
58+
}
59+
},
60+
"pca": {
61+
"algorithm": {
62+
"estimator": "IncrementalPCA",
63+
"library": "sklearnex.preview",
64+
"estimator_methods": {"training": "partial_fit"}
65+
},
66+
"data": {
67+
"split_kwargs": { "test_size": 0.0001 }
68+
}
69+
}
70+
},
71+
"TEMPLATES": {
72+
"basic_statistics": { "SETS": ["common incremental raw gpu params", "basic_statistics", "statistical batches and data"] },
73+
"covariance": { "SETS": ["common incremental raw gpu params", "covariance", "statistical batches and data"] },
74+
"linear_regression": { "SETS": ["common incremental raw gpu params", "linear_regression", "regression batches and data"] },
75+
"pca": { "SETS": ["common incremental raw gpu params", "pca", "statistical batches and data"] }
76+
}
77+
}

‎configs/spmd/large_scale/incremental/basic_stats.json

-30
This file was deleted.

‎configs/spmd/large_scale/incremental/covariance.json

-30
This file was deleted.

‎configs/spmd/large_scale/incremental/linear_model.json

-27
This file was deleted.

‎configs/spmd/large_scale/incremental/pca.json

-30
This file was deleted.

‎configs/spmd/large_scale/spmd_for_online.json

-96
This file was deleted.

0 commit comments

Comments
 (0)