@elaraai/east-py-datascience 0.0.2-beta.8 → 0.0.2-beta.80

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (123) hide show
  1. package/README.md +58 -1
  2. package/dist/src/alns/alns.d.ts +528 -0
  3. package/dist/src/alns/alns.d.ts.map +1 -0
  4. package/dist/src/alns/alns.js +238 -0
  5. package/dist/src/alns/alns.js.map +1 -0
  6. package/dist/src/google_or/google_or.d.ts +2422 -0
  7. package/dist/src/google_or/google_or.d.ts.map +1 -0
  8. package/dist/src/google_or/google_or.js +542 -0
  9. package/dist/src/google_or/google_or.js.map +1 -0
  10. package/dist/{gp → src/gp}/gp.d.ts +185 -136
  11. package/dist/src/gp/gp.d.ts.map +1 -0
  12. package/dist/{gp → src/gp}/gp.js +64 -12
  13. package/dist/src/gp/gp.js.map +1 -0
  14. package/dist/src/index.d.ts +34 -0
  15. package/dist/src/index.d.ts.map +1 -0
  16. package/dist/src/index.js +57 -0
  17. package/dist/src/index.js.map +1 -0
  18. package/dist/src/lightgbm/lightgbm.d.ts +575 -0
  19. package/dist/src/lightgbm/lightgbm.d.ts.map +1 -0
  20. package/dist/{lightgbm → src/lightgbm}/lightgbm.js +104 -18
  21. package/dist/src/lightgbm/lightgbm.js.map +1 -0
  22. package/dist/src/lightning/lightning.d.ts +1594 -0
  23. package/dist/src/lightning/lightning.d.ts.map +1 -0
  24. package/dist/src/lightning/lightning.js +468 -0
  25. package/dist/src/lightning/lightning.js.map +1 -0
  26. package/dist/{mads → src/mads}/mads.d.ts +109 -112
  27. package/dist/src/mads/mads.d.ts.map +1 -0
  28. package/dist/{mads → src/mads}/mads.js +6 -8
  29. package/dist/src/mads/mads.js.map +1 -0
  30. package/dist/src/mapie/mapie.d.ts +3680 -0
  31. package/dist/src/mapie/mapie.d.ts.map +1 -0
  32. package/dist/src/mapie/mapie.js +616 -0
  33. package/dist/src/mapie/mapie.js.map +1 -0
  34. package/dist/{ngboost → src/ngboost}/ngboost.d.ts +192 -142
  35. package/dist/src/ngboost/ngboost.d.ts.map +1 -0
  36. package/dist/{ngboost → src/ngboost}/ngboost.js +67 -14
  37. package/dist/src/ngboost/ngboost.js.map +1 -0
  38. package/dist/src/optimization/optimization.d.ts +420 -0
  39. package/dist/src/optimization/optimization.d.ts.map +1 -0
  40. package/dist/src/optimization/optimization.js +257 -0
  41. package/dist/src/optimization/optimization.js.map +1 -0
  42. package/dist/{optuna → src/optuna}/optuna.d.ts +374 -314
  43. package/dist/src/optuna/optuna.d.ts.map +1 -0
  44. package/dist/{optuna → src/optuna}/optuna.js +2 -0
  45. package/dist/src/optuna/optuna.js.map +1 -0
  46. package/dist/src/pymc/pymc.d.ts +2932 -0
  47. package/dist/src/pymc/pymc.d.ts.map +1 -0
  48. package/dist/src/pymc/pymc.js +688 -0
  49. package/dist/src/pymc/pymc.js.map +1 -0
  50. package/dist/src/scipy/scipy.d.ts +2205 -0
  51. package/dist/src/scipy/scipy.d.ts.map +1 -0
  52. package/dist/src/scipy/scipy.js +884 -0
  53. package/dist/src/scipy/scipy.js.map +1 -0
  54. package/dist/src/shap/shap.d.ts +2988 -0
  55. package/dist/src/shap/shap.d.ts.map +1 -0
  56. package/dist/src/shap/shap.js +500 -0
  57. package/dist/src/shap/shap.js.map +1 -0
  58. package/dist/{simanneal → src/simanneal}/simanneal.d.ts +257 -160
  59. package/dist/src/simanneal/simanneal.d.ts.map +1 -0
  60. package/dist/{simanneal → src/simanneal}/simanneal.js +105 -8
  61. package/dist/src/simanneal/simanneal.js.map +1 -0
  62. package/dist/src/simulation/simulation.d.ts +431 -0
  63. package/dist/src/simulation/simulation.d.ts.map +1 -0
  64. package/dist/src/simulation/simulation.js +306 -0
  65. package/dist/src/simulation/simulation.js.map +1 -0
  66. package/dist/src/sklearn/sklearn.d.ts +6362 -0
  67. package/dist/src/sklearn/sklearn.d.ts.map +1 -0
  68. package/dist/src/sklearn/sklearn.js +1508 -0
  69. package/dist/src/sklearn/sklearn.js.map +1 -0
  70. package/dist/src/torch/torch.d.ts +1205 -0
  71. package/dist/src/torch/torch.d.ts.map +1 -0
  72. package/dist/{torch → src/torch}/torch.js +109 -18
  73. package/dist/src/torch/torch.js.map +1 -0
  74. package/dist/src/types.d.ts +43 -0
  75. package/dist/src/types.d.ts.map +1 -0
  76. package/dist/src/types.js +44 -0
  77. package/dist/src/types.js.map +1 -0
  78. package/dist/src/xgboost/xgboost.d.ts +1424 -0
  79. package/dist/src/xgboost/xgboost.d.ts.map +1 -0
  80. package/dist/src/xgboost/xgboost.js +432 -0
  81. package/dist/src/xgboost/xgboost.js.map +1 -0
  82. package/package.json +12 -12
  83. package/dist/gp/gp.d.ts.map +0 -1
  84. package/dist/gp/gp.js.map +0 -1
  85. package/dist/index.d.ts +0 -27
  86. package/dist/index.d.ts.map +0 -1
  87. package/dist/index.js +0 -41
  88. package/dist/index.js.map +0 -1
  89. package/dist/lightgbm/lightgbm.d.ts +0 -494
  90. package/dist/lightgbm/lightgbm.d.ts.map +0 -1
  91. package/dist/lightgbm/lightgbm.js.map +0 -1
  92. package/dist/mads/mads.d.ts.map +0 -1
  93. package/dist/mads/mads.js.map +0 -1
  94. package/dist/ngboost/ngboost.d.ts.map +0 -1
  95. package/dist/ngboost/ngboost.js.map +0 -1
  96. package/dist/optuna/optuna.d.ts.map +0 -1
  97. package/dist/optuna/optuna.js.map +0 -1
  98. package/dist/scipy/scipy.d.ts +0 -1260
  99. package/dist/scipy/scipy.d.ts.map +0 -1
  100. package/dist/scipy/scipy.js +0 -413
  101. package/dist/scipy/scipy.js.map +0 -1
  102. package/dist/shap/shap.d.ts +0 -657
  103. package/dist/shap/shap.d.ts.map +0 -1
  104. package/dist/shap/shap.js +0 -241
  105. package/dist/shap/shap.js.map +0 -1
  106. package/dist/simanneal/simanneal.d.ts.map +0 -1
  107. package/dist/simanneal/simanneal.js.map +0 -1
  108. package/dist/sklearn/sklearn.d.ts +0 -2691
  109. package/dist/sklearn/sklearn.d.ts.map +0 -1
  110. package/dist/sklearn/sklearn.js +0 -524
  111. package/dist/sklearn/sklearn.js.map +0 -1
  112. package/dist/torch/torch.d.ts +0 -1081
  113. package/dist/torch/torch.d.ts.map +0 -1
  114. package/dist/torch/torch.js.map +0 -1
  115. package/dist/tsconfig.tsbuildinfo +0 -1
  116. package/dist/types.d.ts +0 -80
  117. package/dist/types.d.ts.map +0 -1
  118. package/dist/types.js +0 -81
  119. package/dist/types.js.map +0 -1
  120. package/dist/xgboost/xgboost.d.ts +0 -504
  121. package/dist/xgboost/xgboost.d.ts.map +0 -1
  122. package/dist/xgboost/xgboost.js +0 -177
  123. package/dist/xgboost/xgboost.js.map +0 -1
@@ -1,2691 +0,0 @@
1
- /**
2
- * Copyright (c) 2025 Elara AI Pty Ltd
3
- * Dual-licensed under AGPL-3.0 and commercial license. See LICENSE for details.
4
- */
5
- /**
6
- * Scikit-learn platform functions for East.
7
- *
8
- * Provides core machine learning utilities: preprocessing, model selection, and metrics.
9
- * Uses ONNX for model serialization to enable portable inference.
10
- *
11
- * @packageDocumentation
12
- */
13
- import { StructType, VariantType, OptionType, IntegerType, BooleanType, FloatType, BlobType, ArrayType, StringType, NullType } from "@elaraai/east";
14
- export { VectorType, MatrixType, LabelVectorType } from "../types.js";
15
- export { XGBoostConfigType } from "../xgboost/xgboost.js";
16
- export { LightGBMConfigType } from "../lightgbm/lightgbm.js";
17
- export { NGBoostConfigType } from "../ngboost/ngboost.js";
18
- export { GPConfigType } from "../gp/gp.js";
19
- /**
20
- * Configuration for train/test split.
21
- */
22
- export declare const SplitConfigType: StructType<{
23
- /** Proportion of data to include in test split (default 0.2) */
24
- test_size: OptionType<FloatType>;
25
- /** Random seed for reproducibility */
26
- random_state: OptionType<IntegerType>;
27
- /** Whether to shuffle data before splitting (default true) */
28
- shuffle: OptionType<BooleanType>;
29
- }>;
30
- /**
31
- * Configuration for 3-way train/val/test split.
32
- */
33
- export declare const ThreeWaySplitConfigType: StructType<{
34
- /** Proportion of data for validation (default 0.15) */
35
- val_size: OptionType<FloatType>;
36
- /** Proportion of data for test/holdout (default 0.15) */
37
- test_size: OptionType<FloatType>;
38
- /** Random seed for reproducibility */
39
- random_state: OptionType<IntegerType>;
40
- /** Whether to shuffle data before splitting (default true) */
41
- shuffle: OptionType<BooleanType>;
42
- }>;
43
- /**
44
- * Result of train/test split.
45
- */
46
- export declare const SplitResultType: StructType<{
47
- /** Training features */
48
- X_train: ArrayType<ArrayType<FloatType>>;
49
- /** Test features */
50
- X_test: ArrayType<ArrayType<FloatType>>;
51
- /** Training labels */
52
- y_train: ArrayType<FloatType>;
53
- /** Test labels */
54
- y_test: ArrayType<FloatType>;
55
- }>;
56
- /**
57
- * Result of 3-way train/val/test split.
58
- */
59
- export declare const ThreeWaySplitResultType: StructType<{
60
- /** Training features */
61
- X_train: ArrayType<ArrayType<FloatType>>;
62
- /** Validation features */
63
- X_val: ArrayType<ArrayType<FloatType>>;
64
- /** Test/holdout features */
65
- X_test: ArrayType<ArrayType<FloatType>>;
66
- /** Training targets (matrix) */
67
- Y_train: ArrayType<ArrayType<FloatType>>;
68
- /** Validation targets (matrix) */
69
- Y_val: ArrayType<ArrayType<FloatType>>;
70
- /** Test/holdout targets (matrix) */
71
- Y_test: ArrayType<ArrayType<FloatType>>;
72
- }>;
73
- /**
74
- * Available regression metrics from sklearn.metrics.
75
- */
76
- export declare const RegressionMetricType: VariantType<{
77
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
78
- mse: NullType;
79
- /** Root Mean Squared Error - sqrt(MSE) */
80
- rmse: NullType;
81
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
82
- mae: NullType;
83
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
84
- r2: NullType;
85
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
86
- mape: NullType;
87
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
88
- explained_variance: NullType;
89
- /** Max Error - sklearn.metrics.max_error */
90
- max_error: NullType;
91
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
92
- median_ae: NullType;
93
- }>;
94
- /**
95
- * Single metric result (scalar value).
96
- */
97
- export declare const MetricResultType: StructType<{
98
- /** Which metric was computed */
99
- metric: VariantType<{
100
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
101
- mse: NullType;
102
- /** Root Mean Squared Error - sqrt(MSE) */
103
- rmse: NullType;
104
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
105
- mae: NullType;
106
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
107
- r2: NullType;
108
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
109
- mape: NullType;
110
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
111
- explained_variance: NullType;
112
- /** Max Error - sklearn.metrics.max_error */
113
- max_error: NullType;
114
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
115
- median_ae: NullType;
116
- }>;
117
- /** Scalar metric value */
118
- value: FloatType;
119
- }>;
120
- /**
121
- * Result containing multiple computed metrics.
122
- */
123
- export declare const MetricsResultType: ArrayType<StructType<{
124
- /** Which metric was computed */
125
- metric: VariantType<{
126
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
127
- mse: NullType;
128
- /** Root Mean Squared Error - sqrt(MSE) */
129
- rmse: NullType;
130
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
131
- mae: NullType;
132
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
133
- r2: NullType;
134
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
135
- mape: NullType;
136
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
137
- explained_variance: NullType;
138
- /** Max Error - sklearn.metrics.max_error */
139
- max_error: NullType;
140
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
141
- median_ae: NullType;
142
- }>;
143
- /** Scalar metric value */
144
- value: FloatType;
145
- }>>;
146
- /**
147
- * Aggregation strategy for multi-target metrics.
148
- */
149
- export declare const MetricAggregationType: VariantType<{
150
- /** Return metric for each target separately (default) */
151
- per_target: NullType;
152
- /** Average across all targets (uniform weights) */
153
- uniform_average: NullType;
154
- }>;
155
- /**
156
- * Configuration for multi-target metrics computation.
157
- */
158
- export declare const MultiMetricsConfigType: StructType<{
159
- /** How to aggregate metrics across targets (default: per_target) */
160
- aggregation: OptionType<VariantType<{
161
- /** Return metric for each target separately (default) */
162
- per_target: NullType;
163
- /** Average across all targets (uniform weights) */
164
- uniform_average: NullType;
165
- }>>;
166
- }>;
167
- /**
168
- * Multi-target metric result.
169
- */
170
- export declare const MultiMetricResultType: StructType<{
171
- /** Which metric was computed */
172
- metric: VariantType<{
173
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
174
- mse: NullType;
175
- /** Root Mean Squared Error - sqrt(MSE) */
176
- rmse: NullType;
177
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
178
- mae: NullType;
179
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
180
- r2: NullType;
181
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
182
- mape: NullType;
183
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
184
- explained_variance: NullType;
185
- /** Max Error - sklearn.metrics.max_error */
186
- max_error: NullType;
187
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
188
- median_ae: NullType;
189
- }>;
190
- /** Metric value(s) */
191
- value: VariantType<{
192
- /** Aggregated scalar value */
193
- scalar: FloatType;
194
- /** Per-target values [target_0, target_1, ...] */
195
- per_target: ArrayType<FloatType>;
196
- }>;
197
- }>;
198
- /**
199
- * Result containing multiple computed metrics (multi-target).
200
- */
201
- export declare const MultiMetricsResultType: ArrayType<StructType<{
202
- /** Which metric was computed */
203
- metric: VariantType<{
204
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
205
- mse: NullType;
206
- /** Root Mean Squared Error - sqrt(MSE) */
207
- rmse: NullType;
208
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
209
- mae: NullType;
210
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
211
- r2: NullType;
212
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
213
- mape: NullType;
214
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
215
- explained_variance: NullType;
216
- /** Max Error - sklearn.metrics.max_error */
217
- max_error: NullType;
218
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
219
- median_ae: NullType;
220
- }>;
221
- /** Metric value(s) */
222
- value: VariantType<{
223
- /** Aggregated scalar value */
224
- scalar: FloatType;
225
- /** Per-target values [target_0, target_1, ...] */
226
- per_target: ArrayType<FloatType>;
227
- }>;
228
- }>>;
229
- /**
230
- * Available classification metrics from sklearn.metrics.
231
- */
232
- export declare const ClassificationMetricType: VariantType<{
233
- /** Accuracy - sklearn.metrics.accuracy_score */
234
- accuracy: NullType;
235
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
236
- balanced_accuracy: NullType;
237
- /** Precision - sklearn.metrics.precision_score */
238
- precision: NullType;
239
- /** Recall - sklearn.metrics.recall_score */
240
- recall: NullType;
241
- /** F1 Score - sklearn.metrics.f1_score */
242
- f1: NullType;
243
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
244
- matthews_corrcoef: NullType;
245
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
246
- cohen_kappa: NullType;
247
- /** Jaccard Score - sklearn.metrics.jaccard_score */
248
- jaccard: NullType;
249
- }>;
250
- /**
251
- * Averaging strategy for multi-class classification metrics.
252
- */
253
- export declare const ClassificationAverageType: VariantType<{
254
- /** Calculate metrics for each label, return unweighted mean */
255
- macro: NullType;
256
- /** Calculate metrics globally by counting total TP, FP, FN */
257
- micro: NullType;
258
- /** Calculate metrics for each label, return weighted mean by support */
259
- weighted: NullType;
260
- /** Only for binary classification */
261
- binary: NullType;
262
- }>;
263
- /**
264
- * Configuration for classification metrics.
265
- */
266
- export declare const ClassificationMetricsConfigType: StructType<{
267
- /** Averaging strategy for multi-class (default: macro) */
268
- average: OptionType<VariantType<{
269
- /** Calculate metrics for each label, return unweighted mean */
270
- macro: NullType;
271
- /** Calculate metrics globally by counting total TP, FP, FN */
272
- micro: NullType;
273
- /** Calculate metrics for each label, return weighted mean by support */
274
- weighted: NullType;
275
- /** Only for binary classification */
276
- binary: NullType;
277
- }>>;
278
- }>;
279
- /**
280
- * Single classification metric result.
281
- */
282
- export declare const ClassificationMetricResultType: StructType<{
283
- /** Which metric was computed */
284
- metric: VariantType<{
285
- /** Accuracy - sklearn.metrics.accuracy_score */
286
- accuracy: NullType;
287
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
288
- balanced_accuracy: NullType;
289
- /** Precision - sklearn.metrics.precision_score */
290
- precision: NullType;
291
- /** Recall - sklearn.metrics.recall_score */
292
- recall: NullType;
293
- /** F1 Score - sklearn.metrics.f1_score */
294
- f1: NullType;
295
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
296
- matthews_corrcoef: NullType;
297
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
298
- cohen_kappa: NullType;
299
- /** Jaccard Score - sklearn.metrics.jaccard_score */
300
- jaccard: NullType;
301
- }>;
302
- /** Scalar metric value */
303
- value: FloatType;
304
- }>;
305
- /**
306
- * Result containing multiple computed classification metrics.
307
- */
308
- export declare const ClassificationMetricResultsType: ArrayType<StructType<{
309
- /** Which metric was computed */
310
- metric: VariantType<{
311
- /** Accuracy - sklearn.metrics.accuracy_score */
312
- accuracy: NullType;
313
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
314
- balanced_accuracy: NullType;
315
- /** Precision - sklearn.metrics.precision_score */
316
- precision: NullType;
317
- /** Recall - sklearn.metrics.recall_score */
318
- recall: NullType;
319
- /** F1 Score - sklearn.metrics.f1_score */
320
- f1: NullType;
321
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
322
- matthews_corrcoef: NullType;
323
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
324
- cohen_kappa: NullType;
325
- /** Jaccard Score - sklearn.metrics.jaccard_score */
326
- jaccard: NullType;
327
- }>;
328
- /** Scalar metric value */
329
- value: FloatType;
330
- }>>;
331
- /**
332
- * Configuration for multi-target classification metrics.
333
- */
334
- export declare const MultiClassificationConfigType: StructType<{
335
- /** Averaging strategy for multi-class (default: macro) */
336
- average: OptionType<VariantType<{
337
- /** Calculate metrics for each label, return unweighted mean */
338
- macro: NullType;
339
- /** Calculate metrics globally by counting total TP, FP, FN */
340
- micro: NullType;
341
- /** Calculate metrics for each label, return weighted mean by support */
342
- weighted: NullType;
343
- /** Only for binary classification */
344
- binary: NullType;
345
- }>>;
346
- /** How to aggregate across targets (default: per_target) */
347
- aggregation: OptionType<VariantType<{
348
- /** Return metric for each target separately (default) */
349
- per_target: NullType;
350
- /** Average across all targets (uniform weights) */
351
- uniform_average: NullType;
352
- }>>;
353
- }>;
354
- /**
355
- * Multi-target classification metric result.
356
- */
357
- export declare const MultiClassificationMetricResultType: StructType<{
358
- /** Which metric was computed */
359
- metric: VariantType<{
360
- /** Accuracy - sklearn.metrics.accuracy_score */
361
- accuracy: NullType;
362
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
363
- balanced_accuracy: NullType;
364
- /** Precision - sklearn.metrics.precision_score */
365
- precision: NullType;
366
- /** Recall - sklearn.metrics.recall_score */
367
- recall: NullType;
368
- /** F1 Score - sklearn.metrics.f1_score */
369
- f1: NullType;
370
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
371
- matthews_corrcoef: NullType;
372
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
373
- cohen_kappa: NullType;
374
- /** Jaccard Score - sklearn.metrics.jaccard_score */
375
- jaccard: NullType;
376
- }>;
377
- /** Metric value(s) */
378
- value: VariantType<{
379
- /** Aggregated scalar value */
380
- scalar: FloatType;
381
- /** Per-target values */
382
- per_target: ArrayType<FloatType>;
383
- }>;
384
- }>;
385
- /**
386
- * Result containing multiple computed classification metrics (multi-target).
387
- */
388
- export declare const MultiClassificationMetricResultsType: ArrayType<StructType<{
389
- /** Which metric was computed */
390
- metric: VariantType<{
391
- /** Accuracy - sklearn.metrics.accuracy_score */
392
- accuracy: NullType;
393
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
394
- balanced_accuracy: NullType;
395
- /** Precision - sklearn.metrics.precision_score */
396
- precision: NullType;
397
- /** Recall - sklearn.metrics.recall_score */
398
- recall: NullType;
399
- /** F1 Score - sklearn.metrics.f1_score */
400
- f1: NullType;
401
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
402
- matthews_corrcoef: NullType;
403
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
404
- cohen_kappa: NullType;
405
- /** Jaccard Score - sklearn.metrics.jaccard_score */
406
- jaccard: NullType;
407
- }>;
408
- /** Metric value(s) */
409
- value: VariantType<{
410
- /** Aggregated scalar value */
411
- scalar: FloatType;
412
- /** Per-target values */
413
- per_target: ArrayType<FloatType>;
414
- }>;
415
- }>>;
416
- /**
417
- * Model blob type for serialized sklearn models.
418
- *
419
- * Each model type has its own variant case containing ONNX bytes and metadata.
420
- */
421
- export declare const SklearnModelBlobType: VariantType<{
422
- /** StandardScaler model */
423
- standard_scaler: StructType<{
424
- /** ONNX model bytes */
425
- onnx: BlobType;
426
- /** Number of input features */
427
- n_features: IntegerType;
428
- }>;
429
- /** MinMaxScaler model */
430
- min_max_scaler: StructType<{
431
- /** ONNX model bytes */
432
- onnx: BlobType;
433
- /** Number of input features */
434
- n_features: IntegerType;
435
- }>;
436
- /** RegressorChain model */
437
- regressor_chain: StructType<{
438
- /** Cloudpickle serialized chain */
439
- data: BlobType;
440
- /** Number of input features */
441
- n_features: IntegerType;
442
- /** Number of target outputs */
443
- n_targets: IntegerType;
444
- /** Base estimator type name */
445
- base_estimator_type: StringType;
446
- }>;
447
- }>;
448
- /**
449
- * Base estimator configuration for RegressorChain.
450
- * Variant carries both the estimator type AND its configuration.
451
- */
452
- export declare const RegressorChainBaseConfigType: VariantType<{
453
- /** XGBoost regressor */
454
- xgboost: StructType<{
455
- n_estimators: OptionType<IntegerType>;
456
- max_depth: OptionType<IntegerType>;
457
- learning_rate: OptionType<FloatType>;
458
- min_child_weight: OptionType<IntegerType>;
459
- subsample: OptionType<FloatType>;
460
- colsample_bytree: OptionType<FloatType>;
461
- reg_alpha: OptionType<FloatType>;
462
- reg_lambda: OptionType<FloatType>;
463
- random_state: OptionType<IntegerType>;
464
- n_jobs: OptionType<IntegerType>;
465
- }>;
466
- /** LightGBM regressor */
467
- lightgbm: StructType<{
468
- n_estimators: OptionType<IntegerType>;
469
- max_depth: OptionType<IntegerType>;
470
- learning_rate: OptionType<FloatType>;
471
- num_leaves: OptionType<IntegerType>;
472
- min_child_samples: OptionType<IntegerType>;
473
- subsample: OptionType<FloatType>;
474
- colsample_bytree: OptionType<FloatType>;
475
- reg_alpha: OptionType<FloatType>;
476
- reg_lambda: OptionType<FloatType>;
477
- random_state: OptionType<IntegerType>;
478
- n_jobs: OptionType<IntegerType>;
479
- }>;
480
- /** NGBoost regressor */
481
- ngboost: StructType<{
482
- n_estimators: OptionType<IntegerType>;
483
- learning_rate: OptionType<FloatType>;
484
- minibatch_frac: OptionType<FloatType>;
485
- col_sample: OptionType<FloatType>;
486
- random_state: OptionType<IntegerType>;
487
- distribution: OptionType<VariantType<{
488
- normal: NullType;
489
- lognormal: NullType;
490
- }>>;
491
- }>;
492
- /** Gaussian Process regressor */
493
- gp: StructType<{
494
- kernel: OptionType<VariantType<{
495
- rbf: NullType;
496
- matern_1_2: NullType;
497
- matern_3_2: NullType;
498
- matern_5_2: NullType;
499
- rational_quadratic: NullType;
500
- dot_product: NullType;
501
- }>>;
502
- alpha: OptionType<FloatType>;
503
- n_restarts_optimizer: OptionType<IntegerType>;
504
- normalize_y: OptionType<BooleanType>;
505
- random_state: OptionType<IntegerType>;
506
- }>;
507
- }>;
508
- /**
509
- * Configuration for RegressorChain.
510
- */
511
- export declare const RegressorChainConfigType: StructType<{
512
- /** Base estimator with its configuration */
513
- base_estimator: VariantType<{
514
- /** XGBoost regressor */
515
- xgboost: StructType<{
516
- n_estimators: OptionType<IntegerType>;
517
- max_depth: OptionType<IntegerType>;
518
- learning_rate: OptionType<FloatType>;
519
- min_child_weight: OptionType<IntegerType>;
520
- subsample: OptionType<FloatType>;
521
- colsample_bytree: OptionType<FloatType>;
522
- reg_alpha: OptionType<FloatType>;
523
- reg_lambda: OptionType<FloatType>;
524
- random_state: OptionType<IntegerType>;
525
- n_jobs: OptionType<IntegerType>;
526
- }>;
527
- /** LightGBM regressor */
528
- lightgbm: StructType<{
529
- n_estimators: OptionType<IntegerType>;
530
- max_depth: OptionType<IntegerType>;
531
- learning_rate: OptionType<FloatType>;
532
- num_leaves: OptionType<IntegerType>;
533
- min_child_samples: OptionType<IntegerType>;
534
- subsample: OptionType<FloatType>;
535
- colsample_bytree: OptionType<FloatType>;
536
- reg_alpha: OptionType<FloatType>;
537
- reg_lambda: OptionType<FloatType>;
538
- random_state: OptionType<IntegerType>;
539
- n_jobs: OptionType<IntegerType>;
540
- }>;
541
- /** NGBoost regressor */
542
- ngboost: StructType<{
543
- n_estimators: OptionType<IntegerType>;
544
- learning_rate: OptionType<FloatType>;
545
- minibatch_frac: OptionType<FloatType>;
546
- col_sample: OptionType<FloatType>;
547
- random_state: OptionType<IntegerType>;
548
- distribution: OptionType<VariantType<{
549
- normal: NullType;
550
- lognormal: NullType;
551
- }>>;
552
- }>;
553
- /** Gaussian Process regressor */
554
- gp: StructType<{
555
- kernel: OptionType<VariantType<{
556
- rbf: NullType;
557
- matern_1_2: NullType;
558
- matern_3_2: NullType;
559
- matern_5_2: NullType;
560
- rational_quadratic: NullType;
561
- dot_product: NullType;
562
- }>>;
563
- alpha: OptionType<FloatType>;
564
- n_restarts_optimizer: OptionType<IntegerType>;
565
- normalize_y: OptionType<BooleanType>;
566
- random_state: OptionType<IntegerType>;
567
- }>;
568
- }>;
569
- /** Chain order (indices of targets). None = natural order [0,1,2,...] */
570
- order: OptionType<ArrayType<IntegerType>>;
571
- /** Random seed for reproducibility */
572
- random_state: OptionType<IntegerType>;
573
- }>;
574
- /**
575
- * Split arrays into train and test subsets.
576
- *
577
- * @param X - Feature matrix
578
- * @param y - Target vector
579
- * @param config - Split configuration
580
- * @returns Split result with X_train, X_test, y_train, y_test
581
- */
582
- export declare const sklearn_train_test_split: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<FloatType>, StructType<{
583
- /** Proportion of data to include in test split (default 0.2) */
584
- test_size: OptionType<FloatType>;
585
- /** Random seed for reproducibility */
586
- random_state: OptionType<IntegerType>;
587
- /** Whether to shuffle data before splitting (default true) */
588
- shuffle: OptionType<BooleanType>;
589
- }>], StructType<{
590
- /** Training features */
591
- X_train: ArrayType<ArrayType<FloatType>>;
592
- /** Test features */
593
- X_test: ArrayType<ArrayType<FloatType>>;
594
- /** Training labels */
595
- y_train: ArrayType<FloatType>;
596
- /** Test labels */
597
- y_test: ArrayType<FloatType>;
598
- }>>;
599
- /**
600
- * Fit a StandardScaler to training data.
601
- *
602
- * Standardizes features by removing the mean and scaling to unit variance.
603
- *
604
- * @param X - Training feature matrix
605
- * @returns Model blob containing fitted scaler
606
- */
607
- export declare const sklearn_standard_scaler_fit: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>], VariantType<{
608
- /** StandardScaler model */
609
- standard_scaler: StructType<{
610
- /** ONNX model bytes */
611
- onnx: BlobType;
612
- /** Number of input features */
613
- n_features: IntegerType;
614
- }>;
615
- /** MinMaxScaler model */
616
- min_max_scaler: StructType<{
617
- /** ONNX model bytes */
618
- onnx: BlobType;
619
- /** Number of input features */
620
- n_features: IntegerType;
621
- }>;
622
- /** RegressorChain model */
623
- regressor_chain: StructType<{
624
- /** Cloudpickle serialized chain */
625
- data: BlobType;
626
- /** Number of input features */
627
- n_features: IntegerType;
628
- /** Number of target outputs */
629
- n_targets: IntegerType;
630
- /** Base estimator type name */
631
- base_estimator_type: StringType;
632
- }>;
633
- }>>;
634
- /**
635
- * Transform data using a fitted StandardScaler.
636
- *
637
- * @param model - Fitted scaler model blob
638
- * @param X - Feature matrix to transform
639
- * @returns Transformed feature matrix
640
- */
641
- export declare const sklearn_standard_scaler_transform: import("@elaraai/east").PlatformDefinition<[VariantType<{
642
- /** StandardScaler model */
643
- standard_scaler: StructType<{
644
- /** ONNX model bytes */
645
- onnx: BlobType;
646
- /** Number of input features */
647
- n_features: IntegerType;
648
- }>;
649
- /** MinMaxScaler model */
650
- min_max_scaler: StructType<{
651
- /** ONNX model bytes */
652
- onnx: BlobType;
653
- /** Number of input features */
654
- n_features: IntegerType;
655
- }>;
656
- /** RegressorChain model */
657
- regressor_chain: StructType<{
658
- /** Cloudpickle serialized chain */
659
- data: BlobType;
660
- /** Number of input features */
661
- n_features: IntegerType;
662
- /** Number of target outputs */
663
- n_targets: IntegerType;
664
- /** Base estimator type name */
665
- base_estimator_type: StringType;
666
- }>;
667
- }>, ArrayType<ArrayType<FloatType>>], ArrayType<ArrayType<FloatType>>>;
668
- /**
669
- * Fit a MinMaxScaler to training data.
670
- *
671
- * Scales features to a given range (default [0, 1]).
672
- *
673
- * @param X - Training feature matrix
674
- * @returns Model blob containing fitted scaler
675
- */
676
- export declare const sklearn_min_max_scaler_fit: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>], VariantType<{
677
- /** StandardScaler model */
678
- standard_scaler: StructType<{
679
- /** ONNX model bytes */
680
- onnx: BlobType;
681
- /** Number of input features */
682
- n_features: IntegerType;
683
- }>;
684
- /** MinMaxScaler model */
685
- min_max_scaler: StructType<{
686
- /** ONNX model bytes */
687
- onnx: BlobType;
688
- /** Number of input features */
689
- n_features: IntegerType;
690
- }>;
691
- /** RegressorChain model */
692
- regressor_chain: StructType<{
693
- /** Cloudpickle serialized chain */
694
- data: BlobType;
695
- /** Number of input features */
696
- n_features: IntegerType;
697
- /** Number of target outputs */
698
- n_targets: IntegerType;
699
- /** Base estimator type name */
700
- base_estimator_type: StringType;
701
- }>;
702
- }>>;
703
- /**
704
- * Transform data using a fitted MinMaxScaler.
705
- *
706
- * @param model - Fitted scaler model blob
707
- * @param X - Feature matrix to transform
708
- * @returns Transformed feature matrix
709
- */
710
- export declare const sklearn_min_max_scaler_transform: import("@elaraai/east").PlatformDefinition<[VariantType<{
711
- /** StandardScaler model */
712
- standard_scaler: StructType<{
713
- /** ONNX model bytes */
714
- onnx: BlobType;
715
- /** Number of input features */
716
- n_features: IntegerType;
717
- }>;
718
- /** MinMaxScaler model */
719
- min_max_scaler: StructType<{
720
- /** ONNX model bytes */
721
- onnx: BlobType;
722
- /** Number of input features */
723
- n_features: IntegerType;
724
- }>;
725
- /** RegressorChain model */
726
- regressor_chain: StructType<{
727
- /** Cloudpickle serialized chain */
728
- data: BlobType;
729
- /** Number of input features */
730
- n_features: IntegerType;
731
- /** Number of target outputs */
732
- n_targets: IntegerType;
733
- /** Base estimator type name */
734
- base_estimator_type: StringType;
735
- }>;
736
- }>, ArrayType<ArrayType<FloatType>>], ArrayType<ArrayType<FloatType>>>;
737
- /**
738
- * Train a RegressorChain for multi-target regression.
739
- *
740
- * Each model in the chain uses previous targets as additional features,
741
- * enabling modeling of dependencies between targets.
742
- *
743
- * @param X - Feature matrix
744
- * @param Y - Target matrix (rows=samples, cols=targets)
745
- * @param config - Chain configuration
746
- * @returns Model blob containing fitted chain
747
- */
748
- export declare const sklearn_regressor_chain_train: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, StructType<{
749
- /** Base estimator with its configuration */
750
- base_estimator: VariantType<{
751
- /** XGBoost regressor */
752
- xgboost: StructType<{
753
- n_estimators: OptionType<IntegerType>;
754
- max_depth: OptionType<IntegerType>;
755
- learning_rate: OptionType<FloatType>;
756
- min_child_weight: OptionType<IntegerType>;
757
- subsample: OptionType<FloatType>;
758
- colsample_bytree: OptionType<FloatType>;
759
- reg_alpha: OptionType<FloatType>;
760
- reg_lambda: OptionType<FloatType>;
761
- random_state: OptionType<IntegerType>;
762
- n_jobs: OptionType<IntegerType>;
763
- }>;
764
- /** LightGBM regressor */
765
- lightgbm: StructType<{
766
- n_estimators: OptionType<IntegerType>;
767
- max_depth: OptionType<IntegerType>;
768
- learning_rate: OptionType<FloatType>;
769
- num_leaves: OptionType<IntegerType>;
770
- min_child_samples: OptionType<IntegerType>;
771
- subsample: OptionType<FloatType>;
772
- colsample_bytree: OptionType<FloatType>;
773
- reg_alpha: OptionType<FloatType>;
774
- reg_lambda: OptionType<FloatType>;
775
- random_state: OptionType<IntegerType>;
776
- n_jobs: OptionType<IntegerType>;
777
- }>;
778
- /** NGBoost regressor */
779
- ngboost: StructType<{
780
- n_estimators: OptionType<IntegerType>;
781
- learning_rate: OptionType<FloatType>;
782
- minibatch_frac: OptionType<FloatType>;
783
- col_sample: OptionType<FloatType>;
784
- random_state: OptionType<IntegerType>;
785
- distribution: OptionType<VariantType<{
786
- normal: NullType;
787
- lognormal: NullType;
788
- }>>;
789
- }>;
790
- /** Gaussian Process regressor */
791
- gp: StructType<{
792
- kernel: OptionType<VariantType<{
793
- rbf: NullType;
794
- matern_1_2: NullType;
795
- matern_3_2: NullType;
796
- matern_5_2: NullType;
797
- rational_quadratic: NullType;
798
- dot_product: NullType;
799
- }>>;
800
- alpha: OptionType<FloatType>;
801
- n_restarts_optimizer: OptionType<IntegerType>;
802
- normalize_y: OptionType<BooleanType>;
803
- random_state: OptionType<IntegerType>;
804
- }>;
805
- }>;
806
- /** Chain order (indices of targets). None = natural order [0,1,2,...] */
807
- order: OptionType<ArrayType<IntegerType>>;
808
- /** Random seed for reproducibility */
809
- random_state: OptionType<IntegerType>;
810
- }>], VariantType<{
811
- /** StandardScaler model */
812
- standard_scaler: StructType<{
813
- /** ONNX model bytes */
814
- onnx: BlobType;
815
- /** Number of input features */
816
- n_features: IntegerType;
817
- }>;
818
- /** MinMaxScaler model */
819
- min_max_scaler: StructType<{
820
- /** ONNX model bytes */
821
- onnx: BlobType;
822
- /** Number of input features */
823
- n_features: IntegerType;
824
- }>;
825
- /** RegressorChain model */
826
- regressor_chain: StructType<{
827
- /** Cloudpickle serialized chain */
828
- data: BlobType;
829
- /** Number of input features */
830
- n_features: IntegerType;
831
- /** Number of target outputs */
832
- n_targets: IntegerType;
833
- /** Base estimator type name */
834
- base_estimator_type: StringType;
835
- }>;
836
- }>>;
837
- /**
838
- * Predict using a fitted RegressorChain.
839
- *
840
- * @param model - Fitted chain model blob
841
- * @param X - Feature matrix to predict
842
- * @returns Predicted target matrix
843
- */
844
- export declare const sklearn_regressor_chain_predict: import("@elaraai/east").PlatformDefinition<[VariantType<{
845
- /** StandardScaler model */
846
- standard_scaler: StructType<{
847
- /** ONNX model bytes */
848
- onnx: BlobType;
849
- /** Number of input features */
850
- n_features: IntegerType;
851
- }>;
852
- /** MinMaxScaler model */
853
- min_max_scaler: StructType<{
854
- /** ONNX model bytes */
855
- onnx: BlobType;
856
- /** Number of input features */
857
- n_features: IntegerType;
858
- }>;
859
- /** RegressorChain model */
860
- regressor_chain: StructType<{
861
- /** Cloudpickle serialized chain */
862
- data: BlobType;
863
- /** Number of input features */
864
- n_features: IntegerType;
865
- /** Number of target outputs */
866
- n_targets: IntegerType;
867
- /** Base estimator type name */
868
- base_estimator_type: StringType;
869
- }>;
870
- }>, ArrayType<ArrayType<FloatType>>], ArrayType<ArrayType<FloatType>>>;
871
- /**
872
- * Split arrays into train, validation, and test subsets.
873
- *
874
- * @param X - Feature matrix
875
- * @param Y - Target matrix (multi-target)
876
- * @param config - Split configuration with val_size and test_size
877
- * @returns Split result with X_train, X_val, X_test, Y_train, Y_val, Y_test
878
- */
879
- export declare const sklearn_train_val_test_split: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, StructType<{
880
- /** Proportion of data for validation (default 0.15) */
881
- val_size: OptionType<FloatType>;
882
- /** Proportion of data for test/holdout (default 0.15) */
883
- test_size: OptionType<FloatType>;
884
- /** Random seed for reproducibility */
885
- random_state: OptionType<IntegerType>;
886
- /** Whether to shuffle data before splitting (default true) */
887
- shuffle: OptionType<BooleanType>;
888
- }>], StructType<{
889
- /** Training features */
890
- X_train: ArrayType<ArrayType<FloatType>>;
891
- /** Validation features */
892
- X_val: ArrayType<ArrayType<FloatType>>;
893
- /** Test/holdout features */
894
- X_test: ArrayType<ArrayType<FloatType>>;
895
- /** Training targets (matrix) */
896
- Y_train: ArrayType<ArrayType<FloatType>>;
897
- /** Validation targets (matrix) */
898
- Y_val: ArrayType<ArrayType<FloatType>>;
899
- /** Test/holdout targets (matrix) */
900
- Y_test: ArrayType<ArrayType<FloatType>>;
901
- }>>;
902
- /**
903
- * Compute regression metrics for single-target predictions.
904
- *
905
- * @param y_true - True target values (1D vector)
906
- * @param y_pred - Predicted target values (1D vector)
907
- * @param metrics - Array of metrics to compute
908
- * @returns Array of metric results with scalar values
909
- */
910
- export declare const sklearn_compute_metrics: import("@elaraai/east").PlatformDefinition<[ArrayType<FloatType>, ArrayType<FloatType>, ArrayType<VariantType<{
911
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
912
- mse: NullType;
913
- /** Root Mean Squared Error - sqrt(MSE) */
914
- rmse: NullType;
915
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
916
- mae: NullType;
917
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
918
- r2: NullType;
919
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
920
- mape: NullType;
921
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
922
- explained_variance: NullType;
923
- /** Max Error - sklearn.metrics.max_error */
924
- max_error: NullType;
925
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
926
- median_ae: NullType;
927
- }>>], ArrayType<StructType<{
928
- /** Which metric was computed */
929
- metric: VariantType<{
930
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
931
- mse: NullType;
932
- /** Root Mean Squared Error - sqrt(MSE) */
933
- rmse: NullType;
934
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
935
- mae: NullType;
936
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
937
- r2: NullType;
938
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
939
- mape: NullType;
940
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
941
- explained_variance: NullType;
942
- /** Max Error - sklearn.metrics.max_error */
943
- max_error: NullType;
944
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
945
- median_ae: NullType;
946
- }>;
947
- /** Scalar metric value */
948
- value: FloatType;
949
- }>>>;
950
- /**
951
- * Compute regression metrics for multi-target predictions.
952
- *
953
- * @param Y_true - True target matrix [n_samples, n_targets]
954
- * @param Y_pred - Predicted target matrix [n_samples, n_targets]
955
- * @param metrics - Array of metrics to compute
956
- * @param config - Aggregation configuration
957
- * @returns Array of metric results with per-target or aggregated values
958
- */
959
- export declare const sklearn_compute_metrics_multi: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, ArrayType<VariantType<{
960
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
961
- mse: NullType;
962
- /** Root Mean Squared Error - sqrt(MSE) */
963
- rmse: NullType;
964
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
965
- mae: NullType;
966
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
967
- r2: NullType;
968
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
969
- mape: NullType;
970
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
971
- explained_variance: NullType;
972
- /** Max Error - sklearn.metrics.max_error */
973
- max_error: NullType;
974
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
975
- median_ae: NullType;
976
- }>>, StructType<{
977
- /** How to aggregate metrics across targets (default: per_target) */
978
- aggregation: OptionType<VariantType<{
979
- /** Return metric for each target separately (default) */
980
- per_target: NullType;
981
- /** Average across all targets (uniform weights) */
982
- uniform_average: NullType;
983
- }>>;
984
- }>], ArrayType<StructType<{
985
- /** Which metric was computed */
986
- metric: VariantType<{
987
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
988
- mse: NullType;
989
- /** Root Mean Squared Error - sqrt(MSE) */
990
- rmse: NullType;
991
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
992
- mae: NullType;
993
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
994
- r2: NullType;
995
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
996
- mape: NullType;
997
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
998
- explained_variance: NullType;
999
- /** Max Error - sklearn.metrics.max_error */
1000
- max_error: NullType;
1001
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
1002
- median_ae: NullType;
1003
- }>;
1004
- /** Metric value(s) */
1005
- value: VariantType<{
1006
- /** Aggregated scalar value */
1007
- scalar: FloatType;
1008
- /** Per-target values [target_0, target_1, ...] */
1009
- per_target: ArrayType<FloatType>;
1010
- }>;
1011
- }>>>;
1012
- /**
1013
- * Compute classification metrics for single-target predictions.
1014
- *
1015
- * @param y_true - True class labels (1D integer array)
1016
- * @param y_pred - Predicted class labels (1D integer array)
1017
- * @param metrics - Array of metrics to compute
1018
- * @param config - Configuration (averaging strategy)
1019
- * @returns Array of metric results with scalar values
1020
- */
1021
- export declare const sklearn_compute_classification_metrics: import("@elaraai/east").PlatformDefinition<[ArrayType<IntegerType>, ArrayType<IntegerType>, ArrayType<VariantType<{
1022
- /** Accuracy - sklearn.metrics.accuracy_score */
1023
- accuracy: NullType;
1024
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
1025
- balanced_accuracy: NullType;
1026
- /** Precision - sklearn.metrics.precision_score */
1027
- precision: NullType;
1028
- /** Recall - sklearn.metrics.recall_score */
1029
- recall: NullType;
1030
- /** F1 Score - sklearn.metrics.f1_score */
1031
- f1: NullType;
1032
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
1033
- matthews_corrcoef: NullType;
1034
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
1035
- cohen_kappa: NullType;
1036
- /** Jaccard Score - sklearn.metrics.jaccard_score */
1037
- jaccard: NullType;
1038
- }>>, StructType<{
1039
- /** Averaging strategy for multi-class (default: macro) */
1040
- average: OptionType<VariantType<{
1041
- /** Calculate metrics for each label, return unweighted mean */
1042
- macro: NullType;
1043
- /** Calculate metrics globally by counting total TP, FP, FN */
1044
- micro: NullType;
1045
- /** Calculate metrics for each label, return weighted mean by support */
1046
- weighted: NullType;
1047
- /** Only for binary classification */
1048
- binary: NullType;
1049
- }>>;
1050
- }>], ArrayType<StructType<{
1051
- /** Which metric was computed */
1052
- metric: VariantType<{
1053
- /** Accuracy - sklearn.metrics.accuracy_score */
1054
- accuracy: NullType;
1055
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
1056
- balanced_accuracy: NullType;
1057
- /** Precision - sklearn.metrics.precision_score */
1058
- precision: NullType;
1059
- /** Recall - sklearn.metrics.recall_score */
1060
- recall: NullType;
1061
- /** F1 Score - sklearn.metrics.f1_score */
1062
- f1: NullType;
1063
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
1064
- matthews_corrcoef: NullType;
1065
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
1066
- cohen_kappa: NullType;
1067
- /** Jaccard Score - sklearn.metrics.jaccard_score */
1068
- jaccard: NullType;
1069
- }>;
1070
- /** Scalar metric value */
1071
- value: FloatType;
1072
- }>>>;
1073
- /**
1074
- * Compute classification metrics for multi-target predictions.
1075
- *
1076
- * @param Y_true - True class labels matrix [n_samples, n_targets]
1077
- * @param Y_pred - Predicted class labels matrix [n_samples, n_targets]
1078
- * @param metrics - Array of metrics to compute
1079
- * @param config - Configuration (averaging, aggregation)
1080
- * @returns Array of metric results with per-target or aggregated values
1081
- */
1082
- export declare const sklearn_compute_classification_metrics_multi: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, ArrayType<VariantType<{
1083
- /** Accuracy - sklearn.metrics.accuracy_score */
1084
- accuracy: NullType;
1085
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
1086
- balanced_accuracy: NullType;
1087
- /** Precision - sklearn.metrics.precision_score */
1088
- precision: NullType;
1089
- /** Recall - sklearn.metrics.recall_score */
1090
- recall: NullType;
1091
- /** F1 Score - sklearn.metrics.f1_score */
1092
- f1: NullType;
1093
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
1094
- matthews_corrcoef: NullType;
1095
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
1096
- cohen_kappa: NullType;
1097
- /** Jaccard Score - sklearn.metrics.jaccard_score */
1098
- jaccard: NullType;
1099
- }>>, StructType<{
1100
- /** Averaging strategy for multi-class (default: macro) */
1101
- average: OptionType<VariantType<{
1102
- /** Calculate metrics for each label, return unweighted mean */
1103
- macro: NullType;
1104
- /** Calculate metrics globally by counting total TP, FP, FN */
1105
- micro: NullType;
1106
- /** Calculate metrics for each label, return weighted mean by support */
1107
- weighted: NullType;
1108
- /** Only for binary classification */
1109
- binary: NullType;
1110
- }>>;
1111
- /** How to aggregate across targets (default: per_target) */
1112
- aggregation: OptionType<VariantType<{
1113
- /** Return metric for each target separately (default) */
1114
- per_target: NullType;
1115
- /** Average across all targets (uniform weights) */
1116
- uniform_average: NullType;
1117
- }>>;
1118
- }>], ArrayType<StructType<{
1119
- /** Which metric was computed */
1120
- metric: VariantType<{
1121
- /** Accuracy - sklearn.metrics.accuracy_score */
1122
- accuracy: NullType;
1123
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
1124
- balanced_accuracy: NullType;
1125
- /** Precision - sklearn.metrics.precision_score */
1126
- precision: NullType;
1127
- /** Recall - sklearn.metrics.recall_score */
1128
- recall: NullType;
1129
- /** F1 Score - sklearn.metrics.f1_score */
1130
- f1: NullType;
1131
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
1132
- matthews_corrcoef: NullType;
1133
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
1134
- cohen_kappa: NullType;
1135
- /** Jaccard Score - sklearn.metrics.jaccard_score */
1136
- jaccard: NullType;
1137
- }>;
1138
- /** Metric value(s) */
1139
- value: VariantType<{
1140
- /** Aggregated scalar value */
1141
- scalar: FloatType;
1142
- /** Per-target values */
1143
- per_target: ArrayType<FloatType>;
1144
- }>;
1145
- }>>>;
1146
- /**
1147
- * Type definitions for sklearn functions.
1148
- */
1149
- export declare const SklearnTypes: {
1150
- /** Vector type (array of floats) */
1151
- readonly VectorType: ArrayType<FloatType>;
1152
- /** Matrix type (2D array of floats) */
1153
- readonly MatrixType: ArrayType<ArrayType<FloatType>>;
1154
- /** Label vector type (array of integers) */
1155
- readonly LabelVectorType: ArrayType<IntegerType>;
1156
- /** Split configuration type */
1157
- readonly SplitConfigType: StructType<{
1158
- /** Proportion of data to include in test split (default 0.2) */
1159
- test_size: OptionType<FloatType>;
1160
- /** Random seed for reproducibility */
1161
- random_state: OptionType<IntegerType>;
1162
- /** Whether to shuffle data before splitting (default true) */
1163
- shuffle: OptionType<BooleanType>;
1164
- }>;
1165
- /** Split result type */
1166
- readonly SplitResultType: StructType<{
1167
- /** Training features */
1168
- X_train: ArrayType<ArrayType<FloatType>>;
1169
- /** Test features */
1170
- X_test: ArrayType<ArrayType<FloatType>>;
1171
- /** Training labels */
1172
- y_train: ArrayType<FloatType>;
1173
- /** Test labels */
1174
- y_test: ArrayType<FloatType>;
1175
- }>;
1176
- /** 3-way split configuration type */
1177
- readonly ThreeWaySplitConfigType: StructType<{
1178
- /** Proportion of data for validation (default 0.15) */
1179
- val_size: OptionType<FloatType>;
1180
- /** Proportion of data for test/holdout (default 0.15) */
1181
- test_size: OptionType<FloatType>;
1182
- /** Random seed for reproducibility */
1183
- random_state: OptionType<IntegerType>;
1184
- /** Whether to shuffle data before splitting (default true) */
1185
- shuffle: OptionType<BooleanType>;
1186
- }>;
1187
- /** 3-way split result type */
1188
- readonly ThreeWaySplitResultType: StructType<{
1189
- /** Training features */
1190
- X_train: ArrayType<ArrayType<FloatType>>;
1191
- /** Validation features */
1192
- X_val: ArrayType<ArrayType<FloatType>>;
1193
- /** Test/holdout features */
1194
- X_test: ArrayType<ArrayType<FloatType>>;
1195
- /** Training targets (matrix) */
1196
- Y_train: ArrayType<ArrayType<FloatType>>;
1197
- /** Validation targets (matrix) */
1198
- Y_val: ArrayType<ArrayType<FloatType>>;
1199
- /** Test/holdout targets (matrix) */
1200
- Y_test: ArrayType<ArrayType<FloatType>>;
1201
- }>;
1202
- /** Model blob type for sklearn models */
1203
- readonly ModelBlobType: VariantType<{
1204
- /** StandardScaler model */
1205
- standard_scaler: StructType<{
1206
- /** ONNX model bytes */
1207
- onnx: BlobType;
1208
- /** Number of input features */
1209
- n_features: IntegerType;
1210
- }>;
1211
- /** MinMaxScaler model */
1212
- min_max_scaler: StructType<{
1213
- /** ONNX model bytes */
1214
- onnx: BlobType;
1215
- /** Number of input features */
1216
- n_features: IntegerType;
1217
- }>;
1218
- /** RegressorChain model */
1219
- regressor_chain: StructType<{
1220
- /** Cloudpickle serialized chain */
1221
- data: BlobType;
1222
- /** Number of input features */
1223
- n_features: IntegerType;
1224
- /** Number of target outputs */
1225
- n_targets: IntegerType;
1226
- /** Base estimator type name */
1227
- base_estimator_type: StringType;
1228
- }>;
1229
- }>;
1230
- /** RegressorChain base estimator config type */
1231
- readonly RegressorChainBaseConfigType: VariantType<{
1232
- /** XGBoost regressor */
1233
- xgboost: StructType<{
1234
- n_estimators: OptionType<IntegerType>;
1235
- max_depth: OptionType<IntegerType>;
1236
- learning_rate: OptionType<FloatType>;
1237
- min_child_weight: OptionType<IntegerType>;
1238
- subsample: OptionType<FloatType>;
1239
- colsample_bytree: OptionType<FloatType>;
1240
- reg_alpha: OptionType<FloatType>;
1241
- reg_lambda: OptionType<FloatType>;
1242
- random_state: OptionType<IntegerType>;
1243
- n_jobs: OptionType<IntegerType>;
1244
- }>;
1245
- /** LightGBM regressor */
1246
- lightgbm: StructType<{
1247
- n_estimators: OptionType<IntegerType>;
1248
- max_depth: OptionType<IntegerType>;
1249
- learning_rate: OptionType<FloatType>;
1250
- num_leaves: OptionType<IntegerType>;
1251
- min_child_samples: OptionType<IntegerType>;
1252
- subsample: OptionType<FloatType>;
1253
- colsample_bytree: OptionType<FloatType>;
1254
- reg_alpha: OptionType<FloatType>;
1255
- reg_lambda: OptionType<FloatType>;
1256
- random_state: OptionType<IntegerType>;
1257
- n_jobs: OptionType<IntegerType>;
1258
- }>;
1259
- /** NGBoost regressor */
1260
- ngboost: StructType<{
1261
- n_estimators: OptionType<IntegerType>;
1262
- learning_rate: OptionType<FloatType>;
1263
- minibatch_frac: OptionType<FloatType>;
1264
- col_sample: OptionType<FloatType>;
1265
- random_state: OptionType<IntegerType>;
1266
- distribution: OptionType<VariantType<{
1267
- normal: NullType;
1268
- lognormal: NullType;
1269
- }>>;
1270
- }>;
1271
- /** Gaussian Process regressor */
1272
- gp: StructType<{
1273
- kernel: OptionType<VariantType<{
1274
- rbf: NullType;
1275
- matern_1_2: NullType;
1276
- matern_3_2: NullType;
1277
- matern_5_2: NullType;
1278
- rational_quadratic: NullType;
1279
- dot_product: NullType;
1280
- }>>;
1281
- alpha: OptionType<FloatType>;
1282
- n_restarts_optimizer: OptionType<IntegerType>;
1283
- normalize_y: OptionType<BooleanType>;
1284
- random_state: OptionType<IntegerType>;
1285
- }>;
1286
- }>;
1287
- /** RegressorChain config type */
1288
- readonly RegressorChainConfigType: StructType<{
1289
- /** Base estimator with its configuration */
1290
- base_estimator: VariantType<{
1291
- /** XGBoost regressor */
1292
- xgboost: StructType<{
1293
- n_estimators: OptionType<IntegerType>;
1294
- max_depth: OptionType<IntegerType>;
1295
- learning_rate: OptionType<FloatType>;
1296
- min_child_weight: OptionType<IntegerType>;
1297
- subsample: OptionType<FloatType>;
1298
- colsample_bytree: OptionType<FloatType>;
1299
- reg_alpha: OptionType<FloatType>;
1300
- reg_lambda: OptionType<FloatType>;
1301
- random_state: OptionType<IntegerType>;
1302
- n_jobs: OptionType<IntegerType>;
1303
- }>;
1304
- /** LightGBM regressor */
1305
- lightgbm: StructType<{
1306
- n_estimators: OptionType<IntegerType>;
1307
- max_depth: OptionType<IntegerType>;
1308
- learning_rate: OptionType<FloatType>;
1309
- num_leaves: OptionType<IntegerType>;
1310
- min_child_samples: OptionType<IntegerType>;
1311
- subsample: OptionType<FloatType>;
1312
- colsample_bytree: OptionType<FloatType>;
1313
- reg_alpha: OptionType<FloatType>;
1314
- reg_lambda: OptionType<FloatType>;
1315
- random_state: OptionType<IntegerType>;
1316
- n_jobs: OptionType<IntegerType>;
1317
- }>;
1318
- /** NGBoost regressor */
1319
- ngboost: StructType<{
1320
- n_estimators: OptionType<IntegerType>;
1321
- learning_rate: OptionType<FloatType>;
1322
- minibatch_frac: OptionType<FloatType>;
1323
- col_sample: OptionType<FloatType>;
1324
- random_state: OptionType<IntegerType>;
1325
- distribution: OptionType<VariantType<{
1326
- normal: NullType;
1327
- lognormal: NullType;
1328
- }>>;
1329
- }>;
1330
- /** Gaussian Process regressor */
1331
- gp: StructType<{
1332
- kernel: OptionType<VariantType<{
1333
- rbf: NullType;
1334
- matern_1_2: NullType;
1335
- matern_3_2: NullType;
1336
- matern_5_2: NullType;
1337
- rational_quadratic: NullType;
1338
- dot_product: NullType;
1339
- }>>;
1340
- alpha: OptionType<FloatType>;
1341
- n_restarts_optimizer: OptionType<IntegerType>;
1342
- normalize_y: OptionType<BooleanType>;
1343
- random_state: OptionType<IntegerType>;
1344
- }>;
1345
- }>;
1346
- /** Chain order (indices of targets). None = natural order [0,1,2,...] */
1347
- order: OptionType<ArrayType<IntegerType>>;
1348
- /** Random seed for reproducibility */
1349
- random_state: OptionType<IntegerType>;
1350
- }>;
1351
- /** Regression metric variant */
1352
- readonly RegressionMetricType: VariantType<{
1353
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
1354
- mse: NullType;
1355
- /** Root Mean Squared Error - sqrt(MSE) */
1356
- rmse: NullType;
1357
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
1358
- mae: NullType;
1359
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
1360
- r2: NullType;
1361
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
1362
- mape: NullType;
1363
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
1364
- explained_variance: NullType;
1365
- /** Max Error - sklearn.metrics.max_error */
1366
- max_error: NullType;
1367
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
1368
- median_ae: NullType;
1369
- }>;
1370
- /** Single metric result */
1371
- readonly MetricResultType: StructType<{
1372
- /** Which metric was computed */
1373
- metric: VariantType<{
1374
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
1375
- mse: NullType;
1376
- /** Root Mean Squared Error - sqrt(MSE) */
1377
- rmse: NullType;
1378
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
1379
- mae: NullType;
1380
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
1381
- r2: NullType;
1382
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
1383
- mape: NullType;
1384
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
1385
- explained_variance: NullType;
1386
- /** Max Error - sklearn.metrics.max_error */
1387
- max_error: NullType;
1388
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
1389
- median_ae: NullType;
1390
- }>;
1391
- /** Scalar metric value */
1392
- value: FloatType;
1393
- }>;
1394
- /** Multiple metrics result */
1395
- readonly MetricsResultType: ArrayType<StructType<{
1396
- /** Which metric was computed */
1397
- metric: VariantType<{
1398
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
1399
- mse: NullType;
1400
- /** Root Mean Squared Error - sqrt(MSE) */
1401
- rmse: NullType;
1402
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
1403
- mae: NullType;
1404
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
1405
- r2: NullType;
1406
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
1407
- mape: NullType;
1408
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
1409
- explained_variance: NullType;
1410
- /** Max Error - sklearn.metrics.max_error */
1411
- max_error: NullType;
1412
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
1413
- median_ae: NullType;
1414
- }>;
1415
- /** Scalar metric value */
1416
- value: FloatType;
1417
- }>>;
1418
- /** Metric aggregation type */
1419
- readonly MetricAggregationType: VariantType<{
1420
- /** Return metric for each target separately (default) */
1421
- per_target: NullType;
1422
- /** Average across all targets (uniform weights) */
1423
- uniform_average: NullType;
1424
- }>;
1425
- /** Multi-target metrics config */
1426
- readonly MultiMetricsConfigType: StructType<{
1427
- /** How to aggregate metrics across targets (default: per_target) */
1428
- aggregation: OptionType<VariantType<{
1429
- /** Return metric for each target separately (default) */
1430
- per_target: NullType;
1431
- /** Average across all targets (uniform weights) */
1432
- uniform_average: NullType;
1433
- }>>;
1434
- }>;
1435
- /** Multi-target metric result */
1436
- readonly MultiMetricResultType: StructType<{
1437
- /** Which metric was computed */
1438
- metric: VariantType<{
1439
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
1440
- mse: NullType;
1441
- /** Root Mean Squared Error - sqrt(MSE) */
1442
- rmse: NullType;
1443
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
1444
- mae: NullType;
1445
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
1446
- r2: NullType;
1447
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
1448
- mape: NullType;
1449
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
1450
- explained_variance: NullType;
1451
- /** Max Error - sklearn.metrics.max_error */
1452
- max_error: NullType;
1453
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
1454
- median_ae: NullType;
1455
- }>;
1456
- /** Metric value(s) */
1457
- value: VariantType<{
1458
- /** Aggregated scalar value */
1459
- scalar: FloatType;
1460
- /** Per-target values [target_0, target_1, ...] */
1461
- per_target: ArrayType<FloatType>;
1462
- }>;
1463
- }>;
1464
- /** Multi-target metrics result */
1465
- readonly MultiMetricsResultType: ArrayType<StructType<{
1466
- /** Which metric was computed */
1467
- metric: VariantType<{
1468
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
1469
- mse: NullType;
1470
- /** Root Mean Squared Error - sqrt(MSE) */
1471
- rmse: NullType;
1472
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
1473
- mae: NullType;
1474
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
1475
- r2: NullType;
1476
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
1477
- mape: NullType;
1478
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
1479
- explained_variance: NullType;
1480
- /** Max Error - sklearn.metrics.max_error */
1481
- max_error: NullType;
1482
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
1483
- median_ae: NullType;
1484
- }>;
1485
- /** Metric value(s) */
1486
- value: VariantType<{
1487
- /** Aggregated scalar value */
1488
- scalar: FloatType;
1489
- /** Per-target values [target_0, target_1, ...] */
1490
- per_target: ArrayType<FloatType>;
1491
- }>;
1492
- }>>;
1493
- /** Classification metric variant */
1494
- readonly ClassificationMetricType: VariantType<{
1495
- /** Accuracy - sklearn.metrics.accuracy_score */
1496
- accuracy: NullType;
1497
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
1498
- balanced_accuracy: NullType;
1499
- /** Precision - sklearn.metrics.precision_score */
1500
- precision: NullType;
1501
- /** Recall - sklearn.metrics.recall_score */
1502
- recall: NullType;
1503
- /** F1 Score - sklearn.metrics.f1_score */
1504
- f1: NullType;
1505
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
1506
- matthews_corrcoef: NullType;
1507
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
1508
- cohen_kappa: NullType;
1509
- /** Jaccard Score - sklearn.metrics.jaccard_score */
1510
- jaccard: NullType;
1511
- }>;
1512
- /** Classification averaging type */
1513
- readonly ClassificationAverageType: VariantType<{
1514
- /** Calculate metrics for each label, return unweighted mean */
1515
- macro: NullType;
1516
- /** Calculate metrics globally by counting total TP, FP, FN */
1517
- micro: NullType;
1518
- /** Calculate metrics for each label, return weighted mean by support */
1519
- weighted: NullType;
1520
- /** Only for binary classification */
1521
- binary: NullType;
1522
- }>;
1523
- /** Classification metrics config */
1524
- readonly ClassificationMetricsConfigType: StructType<{
1525
- /** Averaging strategy for multi-class (default: macro) */
1526
- average: OptionType<VariantType<{
1527
- /** Calculate metrics for each label, return unweighted mean */
1528
- macro: NullType;
1529
- /** Calculate metrics globally by counting total TP, FP, FN */
1530
- micro: NullType;
1531
- /** Calculate metrics for each label, return weighted mean by support */
1532
- weighted: NullType;
1533
- /** Only for binary classification */
1534
- binary: NullType;
1535
- }>>;
1536
- }>;
1537
- /** Classification metric result */
1538
- readonly ClassificationMetricResultType: StructType<{
1539
- /** Which metric was computed */
1540
- metric: VariantType<{
1541
- /** Accuracy - sklearn.metrics.accuracy_score */
1542
- accuracy: NullType;
1543
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
1544
- balanced_accuracy: NullType;
1545
- /** Precision - sklearn.metrics.precision_score */
1546
- precision: NullType;
1547
- /** Recall - sklearn.metrics.recall_score */
1548
- recall: NullType;
1549
- /** F1 Score - sklearn.metrics.f1_score */
1550
- f1: NullType;
1551
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
1552
- matthews_corrcoef: NullType;
1553
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
1554
- cohen_kappa: NullType;
1555
- /** Jaccard Score - sklearn.metrics.jaccard_score */
1556
- jaccard: NullType;
1557
- }>;
1558
- /** Scalar metric value */
1559
- value: FloatType;
1560
- }>;
1561
- /** Classification metrics result */
1562
- readonly ClassificationMetricResultsType: ArrayType<StructType<{
1563
- /** Which metric was computed */
1564
- metric: VariantType<{
1565
- /** Accuracy - sklearn.metrics.accuracy_score */
1566
- accuracy: NullType;
1567
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
1568
- balanced_accuracy: NullType;
1569
- /** Precision - sklearn.metrics.precision_score */
1570
- precision: NullType;
1571
- /** Recall - sklearn.metrics.recall_score */
1572
- recall: NullType;
1573
- /** F1 Score - sklearn.metrics.f1_score */
1574
- f1: NullType;
1575
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
1576
- matthews_corrcoef: NullType;
1577
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
1578
- cohen_kappa: NullType;
1579
- /** Jaccard Score - sklearn.metrics.jaccard_score */
1580
- jaccard: NullType;
1581
- }>;
1582
- /** Scalar metric value */
1583
- value: FloatType;
1584
- }>>;
1585
- /** Multi-target classification config */
1586
- readonly MultiClassificationConfigType: StructType<{
1587
- /** Averaging strategy for multi-class (default: macro) */
1588
- average: OptionType<VariantType<{
1589
- /** Calculate metrics for each label, return unweighted mean */
1590
- macro: NullType;
1591
- /** Calculate metrics globally by counting total TP, FP, FN */
1592
- micro: NullType;
1593
- /** Calculate metrics for each label, return weighted mean by support */
1594
- weighted: NullType;
1595
- /** Only for binary classification */
1596
- binary: NullType;
1597
- }>>;
1598
- /** How to aggregate across targets (default: per_target) */
1599
- aggregation: OptionType<VariantType<{
1600
- /** Return metric for each target separately (default) */
1601
- per_target: NullType;
1602
- /** Average across all targets (uniform weights) */
1603
- uniform_average: NullType;
1604
- }>>;
1605
- }>;
1606
- /** Multi-target classification metric result */
1607
- readonly MultiClassificationMetricResultType: StructType<{
1608
- /** Which metric was computed */
1609
- metric: VariantType<{
1610
- /** Accuracy - sklearn.metrics.accuracy_score */
1611
- accuracy: NullType;
1612
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
1613
- balanced_accuracy: NullType;
1614
- /** Precision - sklearn.metrics.precision_score */
1615
- precision: NullType;
1616
- /** Recall - sklearn.metrics.recall_score */
1617
- recall: NullType;
1618
- /** F1 Score - sklearn.metrics.f1_score */
1619
- f1: NullType;
1620
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
1621
- matthews_corrcoef: NullType;
1622
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
1623
- cohen_kappa: NullType;
1624
- /** Jaccard Score - sklearn.metrics.jaccard_score */
1625
- jaccard: NullType;
1626
- }>;
1627
- /** Metric value(s) */
1628
- value: VariantType<{
1629
- /** Aggregated scalar value */
1630
- scalar: FloatType;
1631
- /** Per-target values */
1632
- per_target: ArrayType<FloatType>;
1633
- }>;
1634
- }>;
1635
- /** Multi-target classification metrics result */
1636
- readonly MultiClassificationMetricResultsType: ArrayType<StructType<{
1637
- /** Which metric was computed */
1638
- metric: VariantType<{
1639
- /** Accuracy - sklearn.metrics.accuracy_score */
1640
- accuracy: NullType;
1641
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
1642
- balanced_accuracy: NullType;
1643
- /** Precision - sklearn.metrics.precision_score */
1644
- precision: NullType;
1645
- /** Recall - sklearn.metrics.recall_score */
1646
- recall: NullType;
1647
- /** F1 Score - sklearn.metrics.f1_score */
1648
- f1: NullType;
1649
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
1650
- matthews_corrcoef: NullType;
1651
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
1652
- cohen_kappa: NullType;
1653
- /** Jaccard Score - sklearn.metrics.jaccard_score */
1654
- jaccard: NullType;
1655
- }>;
1656
- /** Metric value(s) */
1657
- value: VariantType<{
1658
- /** Aggregated scalar value */
1659
- scalar: FloatType;
1660
- /** Per-target values */
1661
- per_target: ArrayType<FloatType>;
1662
- }>;
1663
- }>>;
1664
- };
1665
- /**
1666
- * Scikit-learn machine learning utilities.
1667
- *
1668
- * Provides preprocessing, model selection, and metrics for ML workflows.
1669
- *
1670
- * @example
1671
- * ```ts
1672
- * import { East, variant } from "@elaraai/east";
1673
- * import { Sklearn } from "@elaraai/east-py-datascience";
1674
- *
1675
- * const pipeline = East.function([], Sklearn.Types.SplitResultType, $ => {
1676
- * const X = $.let([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0], [7.0, 8.0]]);
1677
- * const y = $.let([1.0, 2.0, 3.0, 4.0]);
1678
- * const config = $.let({
1679
- * test_size: variant('some', 0.25),
1680
- * random_state: variant('some', 42n),
1681
- * shuffle: variant('some', true),
1682
- * });
1683
- * return $.return(Sklearn.trainTestSplit(X, y, config));
1684
- * });
1685
- * ```
1686
- */
1687
- export declare const Sklearn: {
1688
- /** Split arrays into train and test subsets */
1689
- readonly trainTestSplit: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<FloatType>, StructType<{
1690
- /** Proportion of data to include in test split (default 0.2) */
1691
- test_size: OptionType<FloatType>;
1692
- /** Random seed for reproducibility */
1693
- random_state: OptionType<IntegerType>;
1694
- /** Whether to shuffle data before splitting (default true) */
1695
- shuffle: OptionType<BooleanType>;
1696
- }>], StructType<{
1697
- /** Training features */
1698
- X_train: ArrayType<ArrayType<FloatType>>;
1699
- /** Test features */
1700
- X_test: ArrayType<ArrayType<FloatType>>;
1701
- /** Training labels */
1702
- y_train: ArrayType<FloatType>;
1703
- /** Test labels */
1704
- y_test: ArrayType<FloatType>;
1705
- }>>;
1706
- /** Split arrays into train, validation, and test subsets */
1707
- readonly trainValTestSplit: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, StructType<{
1708
- /** Proportion of data for validation (default 0.15) */
1709
- val_size: OptionType<FloatType>;
1710
- /** Proportion of data for test/holdout (default 0.15) */
1711
- test_size: OptionType<FloatType>;
1712
- /** Random seed for reproducibility */
1713
- random_state: OptionType<IntegerType>;
1714
- /** Whether to shuffle data before splitting (default true) */
1715
- shuffle: OptionType<BooleanType>;
1716
- }>], StructType<{
1717
- /** Training features */
1718
- X_train: ArrayType<ArrayType<FloatType>>;
1719
- /** Validation features */
1720
- X_val: ArrayType<ArrayType<FloatType>>;
1721
- /** Test/holdout features */
1722
- X_test: ArrayType<ArrayType<FloatType>>;
1723
- /** Training targets (matrix) */
1724
- Y_train: ArrayType<ArrayType<FloatType>>;
1725
- /** Validation targets (matrix) */
1726
- Y_val: ArrayType<ArrayType<FloatType>>;
1727
- /** Test/holdout targets (matrix) */
1728
- Y_test: ArrayType<ArrayType<FloatType>>;
1729
- }>>;
1730
- /** Fit a StandardScaler to data */
1731
- readonly standardScalerFit: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>], VariantType<{
1732
- /** StandardScaler model */
1733
- standard_scaler: StructType<{
1734
- /** ONNX model bytes */
1735
- onnx: BlobType;
1736
- /** Number of input features */
1737
- n_features: IntegerType;
1738
- }>;
1739
- /** MinMaxScaler model */
1740
- min_max_scaler: StructType<{
1741
- /** ONNX model bytes */
1742
- onnx: BlobType;
1743
- /** Number of input features */
1744
- n_features: IntegerType;
1745
- }>;
1746
- /** RegressorChain model */
1747
- regressor_chain: StructType<{
1748
- /** Cloudpickle serialized chain */
1749
- data: BlobType;
1750
- /** Number of input features */
1751
- n_features: IntegerType;
1752
- /** Number of target outputs */
1753
- n_targets: IntegerType;
1754
- /** Base estimator type name */
1755
- base_estimator_type: StringType;
1756
- }>;
1757
- }>>;
1758
- /** Transform data using fitted StandardScaler */
1759
- readonly standardScalerTransform: import("@elaraai/east").PlatformDefinition<[VariantType<{
1760
- /** StandardScaler model */
1761
- standard_scaler: StructType<{
1762
- /** ONNX model bytes */
1763
- onnx: BlobType;
1764
- /** Number of input features */
1765
- n_features: IntegerType;
1766
- }>;
1767
- /** MinMaxScaler model */
1768
- min_max_scaler: StructType<{
1769
- /** ONNX model bytes */
1770
- onnx: BlobType;
1771
- /** Number of input features */
1772
- n_features: IntegerType;
1773
- }>;
1774
- /** RegressorChain model */
1775
- regressor_chain: StructType<{
1776
- /** Cloudpickle serialized chain */
1777
- data: BlobType;
1778
- /** Number of input features */
1779
- n_features: IntegerType;
1780
- /** Number of target outputs */
1781
- n_targets: IntegerType;
1782
- /** Base estimator type name */
1783
- base_estimator_type: StringType;
1784
- }>;
1785
- }>, ArrayType<ArrayType<FloatType>>], ArrayType<ArrayType<FloatType>>>;
1786
- /** Fit a MinMaxScaler to data */
1787
- readonly minMaxScalerFit: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>], VariantType<{
1788
- /** StandardScaler model */
1789
- standard_scaler: StructType<{
1790
- /** ONNX model bytes */
1791
- onnx: BlobType;
1792
- /** Number of input features */
1793
- n_features: IntegerType;
1794
- }>;
1795
- /** MinMaxScaler model */
1796
- min_max_scaler: StructType<{
1797
- /** ONNX model bytes */
1798
- onnx: BlobType;
1799
- /** Number of input features */
1800
- n_features: IntegerType;
1801
- }>;
1802
- /** RegressorChain model */
1803
- regressor_chain: StructType<{
1804
- /** Cloudpickle serialized chain */
1805
- data: BlobType;
1806
- /** Number of input features */
1807
- n_features: IntegerType;
1808
- /** Number of target outputs */
1809
- n_targets: IntegerType;
1810
- /** Base estimator type name */
1811
- base_estimator_type: StringType;
1812
- }>;
1813
- }>>;
1814
- /** Transform data using fitted MinMaxScaler */
1815
- readonly minMaxScalerTransform: import("@elaraai/east").PlatformDefinition<[VariantType<{
1816
- /** StandardScaler model */
1817
- standard_scaler: StructType<{
1818
- /** ONNX model bytes */
1819
- onnx: BlobType;
1820
- /** Number of input features */
1821
- n_features: IntegerType;
1822
- }>;
1823
- /** MinMaxScaler model */
1824
- min_max_scaler: StructType<{
1825
- /** ONNX model bytes */
1826
- onnx: BlobType;
1827
- /** Number of input features */
1828
- n_features: IntegerType;
1829
- }>;
1830
- /** RegressorChain model */
1831
- regressor_chain: StructType<{
1832
- /** Cloudpickle serialized chain */
1833
- data: BlobType;
1834
- /** Number of input features */
1835
- n_features: IntegerType;
1836
- /** Number of target outputs */
1837
- n_targets: IntegerType;
1838
- /** Base estimator type name */
1839
- base_estimator_type: StringType;
1840
- }>;
1841
- }>, ArrayType<ArrayType<FloatType>>], ArrayType<ArrayType<FloatType>>>;
1842
- /** Compute regression metrics (single-target) */
1843
- readonly computeMetrics: import("@elaraai/east").PlatformDefinition<[ArrayType<FloatType>, ArrayType<FloatType>, ArrayType<VariantType<{
1844
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
1845
- mse: NullType;
1846
- /** Root Mean Squared Error - sqrt(MSE) */
1847
- rmse: NullType;
1848
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
1849
- mae: NullType;
1850
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
1851
- r2: NullType;
1852
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
1853
- mape: NullType;
1854
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
1855
- explained_variance: NullType;
1856
- /** Max Error - sklearn.metrics.max_error */
1857
- max_error: NullType;
1858
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
1859
- median_ae: NullType;
1860
- }>>], ArrayType<StructType<{
1861
- /** Which metric was computed */
1862
- metric: VariantType<{
1863
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
1864
- mse: NullType;
1865
- /** Root Mean Squared Error - sqrt(MSE) */
1866
- rmse: NullType;
1867
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
1868
- mae: NullType;
1869
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
1870
- r2: NullType;
1871
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
1872
- mape: NullType;
1873
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
1874
- explained_variance: NullType;
1875
- /** Max Error - sklearn.metrics.max_error */
1876
- max_error: NullType;
1877
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
1878
- median_ae: NullType;
1879
- }>;
1880
- /** Scalar metric value */
1881
- value: FloatType;
1882
- }>>>;
1883
- /** Compute regression metrics (multi-target) */
1884
- readonly computeMetricsMulti: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, ArrayType<VariantType<{
1885
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
1886
- mse: NullType;
1887
- /** Root Mean Squared Error - sqrt(MSE) */
1888
- rmse: NullType;
1889
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
1890
- mae: NullType;
1891
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
1892
- r2: NullType;
1893
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
1894
- mape: NullType;
1895
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
1896
- explained_variance: NullType;
1897
- /** Max Error - sklearn.metrics.max_error */
1898
- max_error: NullType;
1899
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
1900
- median_ae: NullType;
1901
- }>>, StructType<{
1902
- /** How to aggregate metrics across targets (default: per_target) */
1903
- aggregation: OptionType<VariantType<{
1904
- /** Return metric for each target separately (default) */
1905
- per_target: NullType;
1906
- /** Average across all targets (uniform weights) */
1907
- uniform_average: NullType;
1908
- }>>;
1909
- }>], ArrayType<StructType<{
1910
- /** Which metric was computed */
1911
- metric: VariantType<{
1912
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
1913
- mse: NullType;
1914
- /** Root Mean Squared Error - sqrt(MSE) */
1915
- rmse: NullType;
1916
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
1917
- mae: NullType;
1918
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
1919
- r2: NullType;
1920
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
1921
- mape: NullType;
1922
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
1923
- explained_variance: NullType;
1924
- /** Max Error - sklearn.metrics.max_error */
1925
- max_error: NullType;
1926
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
1927
- median_ae: NullType;
1928
- }>;
1929
- /** Metric value(s) */
1930
- value: VariantType<{
1931
- /** Aggregated scalar value */
1932
- scalar: FloatType;
1933
- /** Per-target values [target_0, target_1, ...] */
1934
- per_target: ArrayType<FloatType>;
1935
- }>;
1936
- }>>>;
1937
- /** Compute classification metrics (single-target) */
1938
- readonly computeClassificationMetrics: import("@elaraai/east").PlatformDefinition<[ArrayType<IntegerType>, ArrayType<IntegerType>, ArrayType<VariantType<{
1939
- /** Accuracy - sklearn.metrics.accuracy_score */
1940
- accuracy: NullType;
1941
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
1942
- balanced_accuracy: NullType;
1943
- /** Precision - sklearn.metrics.precision_score */
1944
- precision: NullType;
1945
- /** Recall - sklearn.metrics.recall_score */
1946
- recall: NullType;
1947
- /** F1 Score - sklearn.metrics.f1_score */
1948
- f1: NullType;
1949
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
1950
- matthews_corrcoef: NullType;
1951
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
1952
- cohen_kappa: NullType;
1953
- /** Jaccard Score - sklearn.metrics.jaccard_score */
1954
- jaccard: NullType;
1955
- }>>, StructType<{
1956
- /** Averaging strategy for multi-class (default: macro) */
1957
- average: OptionType<VariantType<{
1958
- /** Calculate metrics for each label, return unweighted mean */
1959
- macro: NullType;
1960
- /** Calculate metrics globally by counting total TP, FP, FN */
1961
- micro: NullType;
1962
- /** Calculate metrics for each label, return weighted mean by support */
1963
- weighted: NullType;
1964
- /** Only for binary classification */
1965
- binary: NullType;
1966
- }>>;
1967
- }>], ArrayType<StructType<{
1968
- /** Which metric was computed */
1969
- metric: VariantType<{
1970
- /** Accuracy - sklearn.metrics.accuracy_score */
1971
- accuracy: NullType;
1972
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
1973
- balanced_accuracy: NullType;
1974
- /** Precision - sklearn.metrics.precision_score */
1975
- precision: NullType;
1976
- /** Recall - sklearn.metrics.recall_score */
1977
- recall: NullType;
1978
- /** F1 Score - sklearn.metrics.f1_score */
1979
- f1: NullType;
1980
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
1981
- matthews_corrcoef: NullType;
1982
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
1983
- cohen_kappa: NullType;
1984
- /** Jaccard Score - sklearn.metrics.jaccard_score */
1985
- jaccard: NullType;
1986
- }>;
1987
- /** Scalar metric value */
1988
- value: FloatType;
1989
- }>>>;
1990
- /** Compute classification metrics (multi-target) */
1991
- readonly computeClassificationMetricsMulti: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, ArrayType<VariantType<{
1992
- /** Accuracy - sklearn.metrics.accuracy_score */
1993
- accuracy: NullType;
1994
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
1995
- balanced_accuracy: NullType;
1996
- /** Precision - sklearn.metrics.precision_score */
1997
- precision: NullType;
1998
- /** Recall - sklearn.metrics.recall_score */
1999
- recall: NullType;
2000
- /** F1 Score - sklearn.metrics.f1_score */
2001
- f1: NullType;
2002
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
2003
- matthews_corrcoef: NullType;
2004
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
2005
- cohen_kappa: NullType;
2006
- /** Jaccard Score - sklearn.metrics.jaccard_score */
2007
- jaccard: NullType;
2008
- }>>, StructType<{
2009
- /** Averaging strategy for multi-class (default: macro) */
2010
- average: OptionType<VariantType<{
2011
- /** Calculate metrics for each label, return unweighted mean */
2012
- macro: NullType;
2013
- /** Calculate metrics globally by counting total TP, FP, FN */
2014
- micro: NullType;
2015
- /** Calculate metrics for each label, return weighted mean by support */
2016
- weighted: NullType;
2017
- /** Only for binary classification */
2018
- binary: NullType;
2019
- }>>;
2020
- /** How to aggregate across targets (default: per_target) */
2021
- aggregation: OptionType<VariantType<{
2022
- /** Return metric for each target separately (default) */
2023
- per_target: NullType;
2024
- /** Average across all targets (uniform weights) */
2025
- uniform_average: NullType;
2026
- }>>;
2027
- }>], ArrayType<StructType<{
2028
- /** Which metric was computed */
2029
- metric: VariantType<{
2030
- /** Accuracy - sklearn.metrics.accuracy_score */
2031
- accuracy: NullType;
2032
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
2033
- balanced_accuracy: NullType;
2034
- /** Precision - sklearn.metrics.precision_score */
2035
- precision: NullType;
2036
- /** Recall - sklearn.metrics.recall_score */
2037
- recall: NullType;
2038
- /** F1 Score - sklearn.metrics.f1_score */
2039
- f1: NullType;
2040
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
2041
- matthews_corrcoef: NullType;
2042
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
2043
- cohen_kappa: NullType;
2044
- /** Jaccard Score - sklearn.metrics.jaccard_score */
2045
- jaccard: NullType;
2046
- }>;
2047
- /** Metric value(s) */
2048
- value: VariantType<{
2049
- /** Aggregated scalar value */
2050
- scalar: FloatType;
2051
- /** Per-target values */
2052
- per_target: ArrayType<FloatType>;
2053
- }>;
2054
- }>>>;
2055
- /** Train a RegressorChain for multi-target regression */
2056
- readonly regressorChainTrain: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, StructType<{
2057
- /** Base estimator with its configuration */
2058
- base_estimator: VariantType<{
2059
- /** XGBoost regressor */
2060
- xgboost: StructType<{
2061
- n_estimators: OptionType<IntegerType>;
2062
- max_depth: OptionType<IntegerType>;
2063
- learning_rate: OptionType<FloatType>;
2064
- min_child_weight: OptionType<IntegerType>;
2065
- subsample: OptionType<FloatType>;
2066
- colsample_bytree: OptionType<FloatType>;
2067
- reg_alpha: OptionType<FloatType>;
2068
- reg_lambda: OptionType<FloatType>;
2069
- random_state: OptionType<IntegerType>;
2070
- n_jobs: OptionType<IntegerType>;
2071
- }>;
2072
- /** LightGBM regressor */
2073
- lightgbm: StructType<{
2074
- n_estimators: OptionType<IntegerType>;
2075
- max_depth: OptionType<IntegerType>;
2076
- learning_rate: OptionType<FloatType>;
2077
- num_leaves: OptionType<IntegerType>;
2078
- min_child_samples: OptionType<IntegerType>;
2079
- subsample: OptionType<FloatType>;
2080
- colsample_bytree: OptionType<FloatType>;
2081
- reg_alpha: OptionType<FloatType>;
2082
- reg_lambda: OptionType<FloatType>;
2083
- random_state: OptionType<IntegerType>;
2084
- n_jobs: OptionType<IntegerType>;
2085
- }>;
2086
- /** NGBoost regressor */
2087
- ngboost: StructType<{
2088
- n_estimators: OptionType<IntegerType>;
2089
- learning_rate: OptionType<FloatType>;
2090
- minibatch_frac: OptionType<FloatType>;
2091
- col_sample: OptionType<FloatType>;
2092
- random_state: OptionType<IntegerType>;
2093
- distribution: OptionType<VariantType<{
2094
- normal: NullType;
2095
- lognormal: NullType;
2096
- }>>;
2097
- }>;
2098
- /** Gaussian Process regressor */
2099
- gp: StructType<{
2100
- kernel: OptionType<VariantType<{
2101
- rbf: NullType;
2102
- matern_1_2: NullType;
2103
- matern_3_2: NullType;
2104
- matern_5_2: NullType;
2105
- rational_quadratic: NullType;
2106
- dot_product: NullType;
2107
- }>>;
2108
- alpha: OptionType<FloatType>;
2109
- n_restarts_optimizer: OptionType<IntegerType>;
2110
- normalize_y: OptionType<BooleanType>;
2111
- random_state: OptionType<IntegerType>;
2112
- }>;
2113
- }>;
2114
- /** Chain order (indices of targets). None = natural order [0,1,2,...] */
2115
- order: OptionType<ArrayType<IntegerType>>;
2116
- /** Random seed for reproducibility */
2117
- random_state: OptionType<IntegerType>;
2118
- }>], VariantType<{
2119
- /** StandardScaler model */
2120
- standard_scaler: StructType<{
2121
- /** ONNX model bytes */
2122
- onnx: BlobType;
2123
- /** Number of input features */
2124
- n_features: IntegerType;
2125
- }>;
2126
- /** MinMaxScaler model */
2127
- min_max_scaler: StructType<{
2128
- /** ONNX model bytes */
2129
- onnx: BlobType;
2130
- /** Number of input features */
2131
- n_features: IntegerType;
2132
- }>;
2133
- /** RegressorChain model */
2134
- regressor_chain: StructType<{
2135
- /** Cloudpickle serialized chain */
2136
- data: BlobType;
2137
- /** Number of input features */
2138
- n_features: IntegerType;
2139
- /** Number of target outputs */
2140
- n_targets: IntegerType;
2141
- /** Base estimator type name */
2142
- base_estimator_type: StringType;
2143
- }>;
2144
- }>>;
2145
- /** Predict using a fitted RegressorChain */
2146
- readonly regressorChainPredict: import("@elaraai/east").PlatformDefinition<[VariantType<{
2147
- /** StandardScaler model */
2148
- standard_scaler: StructType<{
2149
- /** ONNX model bytes */
2150
- onnx: BlobType;
2151
- /** Number of input features */
2152
- n_features: IntegerType;
2153
- }>;
2154
- /** MinMaxScaler model */
2155
- min_max_scaler: StructType<{
2156
- /** ONNX model bytes */
2157
- onnx: BlobType;
2158
- /** Number of input features */
2159
- n_features: IntegerType;
2160
- }>;
2161
- /** RegressorChain model */
2162
- regressor_chain: StructType<{
2163
- /** Cloudpickle serialized chain */
2164
- data: BlobType;
2165
- /** Number of input features */
2166
- n_features: IntegerType;
2167
- /** Number of target outputs */
2168
- n_targets: IntegerType;
2169
- /** Base estimator type name */
2170
- base_estimator_type: StringType;
2171
- }>;
2172
- }>, ArrayType<ArrayType<FloatType>>], ArrayType<ArrayType<FloatType>>>;
2173
- /** Type definitions */
2174
- readonly Types: {
2175
- /** Vector type (array of floats) */
2176
- readonly VectorType: ArrayType<FloatType>;
2177
- /** Matrix type (2D array of floats) */
2178
- readonly MatrixType: ArrayType<ArrayType<FloatType>>;
2179
- /** Label vector type (array of integers) */
2180
- readonly LabelVectorType: ArrayType<IntegerType>;
2181
- /** Split configuration type */
2182
- readonly SplitConfigType: StructType<{
2183
- /** Proportion of data to include in test split (default 0.2) */
2184
- test_size: OptionType<FloatType>;
2185
- /** Random seed for reproducibility */
2186
- random_state: OptionType<IntegerType>;
2187
- /** Whether to shuffle data before splitting (default true) */
2188
- shuffle: OptionType<BooleanType>;
2189
- }>;
2190
- /** Split result type */
2191
- readonly SplitResultType: StructType<{
2192
- /** Training features */
2193
- X_train: ArrayType<ArrayType<FloatType>>;
2194
- /** Test features */
2195
- X_test: ArrayType<ArrayType<FloatType>>;
2196
- /** Training labels */
2197
- y_train: ArrayType<FloatType>;
2198
- /** Test labels */
2199
- y_test: ArrayType<FloatType>;
2200
- }>;
2201
- /** 3-way split configuration type */
2202
- readonly ThreeWaySplitConfigType: StructType<{
2203
- /** Proportion of data for validation (default 0.15) */
2204
- val_size: OptionType<FloatType>;
2205
- /** Proportion of data for test/holdout (default 0.15) */
2206
- test_size: OptionType<FloatType>;
2207
- /** Random seed for reproducibility */
2208
- random_state: OptionType<IntegerType>;
2209
- /** Whether to shuffle data before splitting (default true) */
2210
- shuffle: OptionType<BooleanType>;
2211
- }>;
2212
- /** 3-way split result type */
2213
- readonly ThreeWaySplitResultType: StructType<{
2214
- /** Training features */
2215
- X_train: ArrayType<ArrayType<FloatType>>;
2216
- /** Validation features */
2217
- X_val: ArrayType<ArrayType<FloatType>>;
2218
- /** Test/holdout features */
2219
- X_test: ArrayType<ArrayType<FloatType>>;
2220
- /** Training targets (matrix) */
2221
- Y_train: ArrayType<ArrayType<FloatType>>;
2222
- /** Validation targets (matrix) */
2223
- Y_val: ArrayType<ArrayType<FloatType>>;
2224
- /** Test/holdout targets (matrix) */
2225
- Y_test: ArrayType<ArrayType<FloatType>>;
2226
- }>;
2227
- /** Model blob type for sklearn models */
2228
- readonly ModelBlobType: VariantType<{
2229
- /** StandardScaler model */
2230
- standard_scaler: StructType<{
2231
- /** ONNX model bytes */
2232
- onnx: BlobType;
2233
- /** Number of input features */
2234
- n_features: IntegerType;
2235
- }>;
2236
- /** MinMaxScaler model */
2237
- min_max_scaler: StructType<{
2238
- /** ONNX model bytes */
2239
- onnx: BlobType;
2240
- /** Number of input features */
2241
- n_features: IntegerType;
2242
- }>;
2243
- /** RegressorChain model */
2244
- regressor_chain: StructType<{
2245
- /** Cloudpickle serialized chain */
2246
- data: BlobType;
2247
- /** Number of input features */
2248
- n_features: IntegerType;
2249
- /** Number of target outputs */
2250
- n_targets: IntegerType;
2251
- /** Base estimator type name */
2252
- base_estimator_type: StringType;
2253
- }>;
2254
- }>;
2255
- /** RegressorChain base estimator config type */
2256
- readonly RegressorChainBaseConfigType: VariantType<{
2257
- /** XGBoost regressor */
2258
- xgboost: StructType<{
2259
- n_estimators: OptionType<IntegerType>;
2260
- max_depth: OptionType<IntegerType>;
2261
- learning_rate: OptionType<FloatType>;
2262
- min_child_weight: OptionType<IntegerType>;
2263
- subsample: OptionType<FloatType>;
2264
- colsample_bytree: OptionType<FloatType>;
2265
- reg_alpha: OptionType<FloatType>;
2266
- reg_lambda: OptionType<FloatType>;
2267
- random_state: OptionType<IntegerType>;
2268
- n_jobs: OptionType<IntegerType>;
2269
- }>;
2270
- /** LightGBM regressor */
2271
- lightgbm: StructType<{
2272
- n_estimators: OptionType<IntegerType>;
2273
- max_depth: OptionType<IntegerType>;
2274
- learning_rate: OptionType<FloatType>;
2275
- num_leaves: OptionType<IntegerType>;
2276
- min_child_samples: OptionType<IntegerType>;
2277
- subsample: OptionType<FloatType>;
2278
- colsample_bytree: OptionType<FloatType>;
2279
- reg_alpha: OptionType<FloatType>;
2280
- reg_lambda: OptionType<FloatType>;
2281
- random_state: OptionType<IntegerType>;
2282
- n_jobs: OptionType<IntegerType>;
2283
- }>;
2284
- /** NGBoost regressor */
2285
- ngboost: StructType<{
2286
- n_estimators: OptionType<IntegerType>;
2287
- learning_rate: OptionType<FloatType>;
2288
- minibatch_frac: OptionType<FloatType>;
2289
- col_sample: OptionType<FloatType>;
2290
- random_state: OptionType<IntegerType>;
2291
- distribution: OptionType<VariantType<{
2292
- normal: NullType;
2293
- lognormal: NullType;
2294
- }>>;
2295
- }>;
2296
- /** Gaussian Process regressor */
2297
- gp: StructType<{
2298
- kernel: OptionType<VariantType<{
2299
- rbf: NullType;
2300
- matern_1_2: NullType;
2301
- matern_3_2: NullType;
2302
- matern_5_2: NullType;
2303
- rational_quadratic: NullType;
2304
- dot_product: NullType;
2305
- }>>;
2306
- alpha: OptionType<FloatType>;
2307
- n_restarts_optimizer: OptionType<IntegerType>;
2308
- normalize_y: OptionType<BooleanType>;
2309
- random_state: OptionType<IntegerType>;
2310
- }>;
2311
- }>;
2312
- /** RegressorChain config type */
2313
- readonly RegressorChainConfigType: StructType<{
2314
- /** Base estimator with its configuration */
2315
- base_estimator: VariantType<{
2316
- /** XGBoost regressor */
2317
- xgboost: StructType<{
2318
- n_estimators: OptionType<IntegerType>;
2319
- max_depth: OptionType<IntegerType>;
2320
- learning_rate: OptionType<FloatType>;
2321
- min_child_weight: OptionType<IntegerType>;
2322
- subsample: OptionType<FloatType>;
2323
- colsample_bytree: OptionType<FloatType>;
2324
- reg_alpha: OptionType<FloatType>;
2325
- reg_lambda: OptionType<FloatType>;
2326
- random_state: OptionType<IntegerType>;
2327
- n_jobs: OptionType<IntegerType>;
2328
- }>;
2329
- /** LightGBM regressor */
2330
- lightgbm: StructType<{
2331
- n_estimators: OptionType<IntegerType>;
2332
- max_depth: OptionType<IntegerType>;
2333
- learning_rate: OptionType<FloatType>;
2334
- num_leaves: OptionType<IntegerType>;
2335
- min_child_samples: OptionType<IntegerType>;
2336
- subsample: OptionType<FloatType>;
2337
- colsample_bytree: OptionType<FloatType>;
2338
- reg_alpha: OptionType<FloatType>;
2339
- reg_lambda: OptionType<FloatType>;
2340
- random_state: OptionType<IntegerType>;
2341
- n_jobs: OptionType<IntegerType>;
2342
- }>;
2343
- /** NGBoost regressor */
2344
- ngboost: StructType<{
2345
- n_estimators: OptionType<IntegerType>;
2346
- learning_rate: OptionType<FloatType>;
2347
- minibatch_frac: OptionType<FloatType>;
2348
- col_sample: OptionType<FloatType>;
2349
- random_state: OptionType<IntegerType>;
2350
- distribution: OptionType<VariantType<{
2351
- normal: NullType;
2352
- lognormal: NullType;
2353
- }>>;
2354
- }>;
2355
- /** Gaussian Process regressor */
2356
- gp: StructType<{
2357
- kernel: OptionType<VariantType<{
2358
- rbf: NullType;
2359
- matern_1_2: NullType;
2360
- matern_3_2: NullType;
2361
- matern_5_2: NullType;
2362
- rational_quadratic: NullType;
2363
- dot_product: NullType;
2364
- }>>;
2365
- alpha: OptionType<FloatType>;
2366
- n_restarts_optimizer: OptionType<IntegerType>;
2367
- normalize_y: OptionType<BooleanType>;
2368
- random_state: OptionType<IntegerType>;
2369
- }>;
2370
- }>;
2371
- /** Chain order (indices of targets). None = natural order [0,1,2,...] */
2372
- order: OptionType<ArrayType<IntegerType>>;
2373
- /** Random seed for reproducibility */
2374
- random_state: OptionType<IntegerType>;
2375
- }>;
2376
- /** Regression metric variant */
2377
- readonly RegressionMetricType: VariantType<{
2378
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
2379
- mse: NullType;
2380
- /** Root Mean Squared Error - sqrt(MSE) */
2381
- rmse: NullType;
2382
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
2383
- mae: NullType;
2384
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
2385
- r2: NullType;
2386
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
2387
- mape: NullType;
2388
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
2389
- explained_variance: NullType;
2390
- /** Max Error - sklearn.metrics.max_error */
2391
- max_error: NullType;
2392
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
2393
- median_ae: NullType;
2394
- }>;
2395
- /** Single metric result */
2396
- readonly MetricResultType: StructType<{
2397
- /** Which metric was computed */
2398
- metric: VariantType<{
2399
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
2400
- mse: NullType;
2401
- /** Root Mean Squared Error - sqrt(MSE) */
2402
- rmse: NullType;
2403
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
2404
- mae: NullType;
2405
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
2406
- r2: NullType;
2407
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
2408
- mape: NullType;
2409
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
2410
- explained_variance: NullType;
2411
- /** Max Error - sklearn.metrics.max_error */
2412
- max_error: NullType;
2413
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
2414
- median_ae: NullType;
2415
- }>;
2416
- /** Scalar metric value */
2417
- value: FloatType;
2418
- }>;
2419
- /** Multiple metrics result */
2420
- readonly MetricsResultType: ArrayType<StructType<{
2421
- /** Which metric was computed */
2422
- metric: VariantType<{
2423
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
2424
- mse: NullType;
2425
- /** Root Mean Squared Error - sqrt(MSE) */
2426
- rmse: NullType;
2427
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
2428
- mae: NullType;
2429
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
2430
- r2: NullType;
2431
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
2432
- mape: NullType;
2433
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
2434
- explained_variance: NullType;
2435
- /** Max Error - sklearn.metrics.max_error */
2436
- max_error: NullType;
2437
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
2438
- median_ae: NullType;
2439
- }>;
2440
- /** Scalar metric value */
2441
- value: FloatType;
2442
- }>>;
2443
- /** Metric aggregation type */
2444
- readonly MetricAggregationType: VariantType<{
2445
- /** Return metric for each target separately (default) */
2446
- per_target: NullType;
2447
- /** Average across all targets (uniform weights) */
2448
- uniform_average: NullType;
2449
- }>;
2450
- /** Multi-target metrics config */
2451
- readonly MultiMetricsConfigType: StructType<{
2452
- /** How to aggregate metrics across targets (default: per_target) */
2453
- aggregation: OptionType<VariantType<{
2454
- /** Return metric for each target separately (default) */
2455
- per_target: NullType;
2456
- /** Average across all targets (uniform weights) */
2457
- uniform_average: NullType;
2458
- }>>;
2459
- }>;
2460
- /** Multi-target metric result */
2461
- readonly MultiMetricResultType: StructType<{
2462
- /** Which metric was computed */
2463
- metric: VariantType<{
2464
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
2465
- mse: NullType;
2466
- /** Root Mean Squared Error - sqrt(MSE) */
2467
- rmse: NullType;
2468
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
2469
- mae: NullType;
2470
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
2471
- r2: NullType;
2472
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
2473
- mape: NullType;
2474
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
2475
- explained_variance: NullType;
2476
- /** Max Error - sklearn.metrics.max_error */
2477
- max_error: NullType;
2478
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
2479
- median_ae: NullType;
2480
- }>;
2481
- /** Metric value(s) */
2482
- value: VariantType<{
2483
- /** Aggregated scalar value */
2484
- scalar: FloatType;
2485
- /** Per-target values [target_0, target_1, ...] */
2486
- per_target: ArrayType<FloatType>;
2487
- }>;
2488
- }>;
2489
- /** Multi-target metrics result */
2490
- readonly MultiMetricsResultType: ArrayType<StructType<{
2491
- /** Which metric was computed */
2492
- metric: VariantType<{
2493
- /** Mean Squared Error - sklearn.metrics.mean_squared_error */
2494
- mse: NullType;
2495
- /** Root Mean Squared Error - sqrt(MSE) */
2496
- rmse: NullType;
2497
- /** Mean Absolute Error - sklearn.metrics.mean_absolute_error */
2498
- mae: NullType;
2499
- /** R² (coefficient of determination) - sklearn.metrics.r2_score */
2500
- r2: NullType;
2501
- /** Mean Absolute Percentage Error - sklearn.metrics.mean_absolute_percentage_error */
2502
- mape: NullType;
2503
- /** Explained Variance Score - sklearn.metrics.explained_variance_score */
2504
- explained_variance: NullType;
2505
- /** Max Error - sklearn.metrics.max_error */
2506
- max_error: NullType;
2507
- /** Median Absolute Error - sklearn.metrics.median_absolute_error */
2508
- median_ae: NullType;
2509
- }>;
2510
- /** Metric value(s) */
2511
- value: VariantType<{
2512
- /** Aggregated scalar value */
2513
- scalar: FloatType;
2514
- /** Per-target values [target_0, target_1, ...] */
2515
- per_target: ArrayType<FloatType>;
2516
- }>;
2517
- }>>;
2518
- /** Classification metric variant */
2519
- readonly ClassificationMetricType: VariantType<{
2520
- /** Accuracy - sklearn.metrics.accuracy_score */
2521
- accuracy: NullType;
2522
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
2523
- balanced_accuracy: NullType;
2524
- /** Precision - sklearn.metrics.precision_score */
2525
- precision: NullType;
2526
- /** Recall - sklearn.metrics.recall_score */
2527
- recall: NullType;
2528
- /** F1 Score - sklearn.metrics.f1_score */
2529
- f1: NullType;
2530
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
2531
- matthews_corrcoef: NullType;
2532
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
2533
- cohen_kappa: NullType;
2534
- /** Jaccard Score - sklearn.metrics.jaccard_score */
2535
- jaccard: NullType;
2536
- }>;
2537
- /** Classification averaging type */
2538
- readonly ClassificationAverageType: VariantType<{
2539
- /** Calculate metrics for each label, return unweighted mean */
2540
- macro: NullType;
2541
- /** Calculate metrics globally by counting total TP, FP, FN */
2542
- micro: NullType;
2543
- /** Calculate metrics for each label, return weighted mean by support */
2544
- weighted: NullType;
2545
- /** Only for binary classification */
2546
- binary: NullType;
2547
- }>;
2548
- /** Classification metrics config */
2549
- readonly ClassificationMetricsConfigType: StructType<{
2550
- /** Averaging strategy for multi-class (default: macro) */
2551
- average: OptionType<VariantType<{
2552
- /** Calculate metrics for each label, return unweighted mean */
2553
- macro: NullType;
2554
- /** Calculate metrics globally by counting total TP, FP, FN */
2555
- micro: NullType;
2556
- /** Calculate metrics for each label, return weighted mean by support */
2557
- weighted: NullType;
2558
- /** Only for binary classification */
2559
- binary: NullType;
2560
- }>>;
2561
- }>;
2562
- /** Classification metric result */
2563
- readonly ClassificationMetricResultType: StructType<{
2564
- /** Which metric was computed */
2565
- metric: VariantType<{
2566
- /** Accuracy - sklearn.metrics.accuracy_score */
2567
- accuracy: NullType;
2568
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
2569
- balanced_accuracy: NullType;
2570
- /** Precision - sklearn.metrics.precision_score */
2571
- precision: NullType;
2572
- /** Recall - sklearn.metrics.recall_score */
2573
- recall: NullType;
2574
- /** F1 Score - sklearn.metrics.f1_score */
2575
- f1: NullType;
2576
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
2577
- matthews_corrcoef: NullType;
2578
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
2579
- cohen_kappa: NullType;
2580
- /** Jaccard Score - sklearn.metrics.jaccard_score */
2581
- jaccard: NullType;
2582
- }>;
2583
- /** Scalar metric value */
2584
- value: FloatType;
2585
- }>;
2586
- /** Classification metrics result */
2587
- readonly ClassificationMetricResultsType: ArrayType<StructType<{
2588
- /** Which metric was computed */
2589
- metric: VariantType<{
2590
- /** Accuracy - sklearn.metrics.accuracy_score */
2591
- accuracy: NullType;
2592
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
2593
- balanced_accuracy: NullType;
2594
- /** Precision - sklearn.metrics.precision_score */
2595
- precision: NullType;
2596
- /** Recall - sklearn.metrics.recall_score */
2597
- recall: NullType;
2598
- /** F1 Score - sklearn.metrics.f1_score */
2599
- f1: NullType;
2600
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
2601
- matthews_corrcoef: NullType;
2602
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
2603
- cohen_kappa: NullType;
2604
- /** Jaccard Score - sklearn.metrics.jaccard_score */
2605
- jaccard: NullType;
2606
- }>;
2607
- /** Scalar metric value */
2608
- value: FloatType;
2609
- }>>;
2610
- /** Multi-target classification config */
2611
- readonly MultiClassificationConfigType: StructType<{
2612
- /** Averaging strategy for multi-class (default: macro) */
2613
- average: OptionType<VariantType<{
2614
- /** Calculate metrics for each label, return unweighted mean */
2615
- macro: NullType;
2616
- /** Calculate metrics globally by counting total TP, FP, FN */
2617
- micro: NullType;
2618
- /** Calculate metrics for each label, return weighted mean by support */
2619
- weighted: NullType;
2620
- /** Only for binary classification */
2621
- binary: NullType;
2622
- }>>;
2623
- /** How to aggregate across targets (default: per_target) */
2624
- aggregation: OptionType<VariantType<{
2625
- /** Return metric for each target separately (default) */
2626
- per_target: NullType;
2627
- /** Average across all targets (uniform weights) */
2628
- uniform_average: NullType;
2629
- }>>;
2630
- }>;
2631
- /** Multi-target classification metric result */
2632
- readonly MultiClassificationMetricResultType: StructType<{
2633
- /** Which metric was computed */
2634
- metric: VariantType<{
2635
- /** Accuracy - sklearn.metrics.accuracy_score */
2636
- accuracy: NullType;
2637
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
2638
- balanced_accuracy: NullType;
2639
- /** Precision - sklearn.metrics.precision_score */
2640
- precision: NullType;
2641
- /** Recall - sklearn.metrics.recall_score */
2642
- recall: NullType;
2643
- /** F1 Score - sklearn.metrics.f1_score */
2644
- f1: NullType;
2645
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
2646
- matthews_corrcoef: NullType;
2647
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
2648
- cohen_kappa: NullType;
2649
- /** Jaccard Score - sklearn.metrics.jaccard_score */
2650
- jaccard: NullType;
2651
- }>;
2652
- /** Metric value(s) */
2653
- value: VariantType<{
2654
- /** Aggregated scalar value */
2655
- scalar: FloatType;
2656
- /** Per-target values */
2657
- per_target: ArrayType<FloatType>;
2658
- }>;
2659
- }>;
2660
- /** Multi-target classification metrics result */
2661
- readonly MultiClassificationMetricResultsType: ArrayType<StructType<{
2662
- /** Which metric was computed */
2663
- metric: VariantType<{
2664
- /** Accuracy - sklearn.metrics.accuracy_score */
2665
- accuracy: NullType;
2666
- /** Balanced Accuracy - sklearn.metrics.balanced_accuracy_score */
2667
- balanced_accuracy: NullType;
2668
- /** Precision - sklearn.metrics.precision_score */
2669
- precision: NullType;
2670
- /** Recall - sklearn.metrics.recall_score */
2671
- recall: NullType;
2672
- /** F1 Score - sklearn.metrics.f1_score */
2673
- f1: NullType;
2674
- /** Matthews Correlation Coefficient - sklearn.metrics.matthews_corrcoef */
2675
- matthews_corrcoef: NullType;
2676
- /** Cohen's Kappa - sklearn.metrics.cohen_kappa_score */
2677
- cohen_kappa: NullType;
2678
- /** Jaccard Score - sklearn.metrics.jaccard_score */
2679
- jaccard: NullType;
2680
- }>;
2681
- /** Metric value(s) */
2682
- value: VariantType<{
2683
- /** Aggregated scalar value */
2684
- scalar: FloatType;
2685
- /** Per-target values */
2686
- per_target: ArrayType<FloatType>;
2687
- }>;
2688
- }>>;
2689
- };
2690
- };
2691
- //# sourceMappingURL=sklearn.d.ts.map