|
47 | 47 | },
|
48 | 48 | "split_kwargs": {"ignore": true}
|
49 | 49 | }
|
| 50 | + }, |
| 51 | + "basic_statistics": { |
| 52 | + "algorithm": [ |
| 53 | + { |
| 54 | + "estimator": "BasicStatistics", |
| 55 | + "library": "sklearnex.spmd", |
| 56 | + "estimator_methods": {"training": "fit"} |
| 57 | + } |
| 58 | + ] |
| 59 | + }, |
| 60 | + "covariance": { |
| 61 | + "algorithm": [ |
| 62 | + { |
| 63 | + "estimator": "EmpiricalCovariance", |
| 64 | + "library": "sklearnex.spmd.covariance", |
| 65 | + "estimator_methods": {"training": "fit"} |
| 66 | + } |
| 67 | + ] |
| 68 | + }, |
| 69 | + "linear_regression": { |
| 70 | + "algorithm": [ |
| 71 | + { |
| 72 | + "estimator": "LinearRegression", |
| 73 | + "library": "sklearnex.spmd.linear_model", |
| 74 | + "estimator_methods": {"training": "fit"} |
| 75 | + } |
| 76 | + ] |
| 77 | + }, |
| 78 | + "pca": { |
| 79 | + "algorithm": [ |
| 80 | + { |
| 81 | + "estimator": "PCA", |
| 82 | + "library": "sklearnex.spmd.decomposition", |
| 83 | + "estimator_methods": {"training": "fit"} |
| 84 | + } |
| 85 | + ] |
50 | 86 | }
|
51 | 87 | },
|
52 | 88 | "TEMPLATES": {
|
53 |
| - "basic_statistics": {"SETS": ["basic_statistics data", "sklearnex spmd implementation", "large scale full one node parameters"]}, |
54 |
| - "covariance": {"SETS": ["covariance data", "sklearnex spmd implementation", "large scale full one node parameters"]}, |
| 89 | + "basic_statistics": {"SETS": ["basic_statistics", "basic_statistics data", "sklearnex spmd implementation", "large scale full one node parameters"]}, |
| 90 | + "covariance": {"SETS": ["covariance", "covariance data", "sklearnex spmd implementation", "large scale full one node parameters"]}, |
55 | 91 | "linear_regression": {
|
56 |
| - "SETS": ["linear_regression data", "sklearnex spmd implementation", "large scale full one node parameters"] |
| 92 | + "SETS": ["linear_regression", "linear_regression data", "sklearnex spmd implementation", "large scale full one node parameters"] |
57 | 93 | },
|
58 |
| - "pca": {"SETS": ["pca data", "sklearnex spmd implementation", "large scale full one node parameters"]} |
| 94 | + "pca": {"SETS": ["pca", "pca data", "sklearnex spmd implementation", "large scale full one node parameters"]} |
59 | 95 | }
|
60 | 96 | }
|
0 commit comments