@elaraai/east-py-datascience 1.0.1 → 1.0.2

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 (81) hide show
  1. package/dist/src/alns/alns.d.ts +528 -0
  2. package/dist/src/alns/alns.d.ts.map +1 -0
  3. package/dist/src/alns/alns.js +238 -0
  4. package/dist/src/alns/alns.js.map +1 -0
  5. package/dist/src/google_or/google_or.d.ts +2457 -0
  6. package/dist/src/google_or/google_or.d.ts.map +1 -0
  7. package/dist/src/google_or/google_or.js +547 -0
  8. package/dist/src/google_or/google_or.js.map +1 -0
  9. package/dist/src/gp/gp.d.ts +447 -0
  10. package/dist/src/gp/gp.d.ts.map +1 -0
  11. package/dist/src/gp/gp.js +222 -0
  12. package/dist/src/gp/gp.js.map +1 -0
  13. package/dist/src/index.d.ts +34 -0
  14. package/dist/src/index.d.ts.map +1 -0
  15. package/dist/src/index.js +57 -0
  16. package/dist/src/index.js.map +1 -0
  17. package/dist/src/lightgbm/lightgbm.d.ts +575 -0
  18. package/dist/src/lightgbm/lightgbm.d.ts.map +1 -0
  19. package/dist/src/lightgbm/lightgbm.js +241 -0
  20. package/dist/src/lightgbm/lightgbm.js.map +1 -0
  21. package/dist/src/lightning/lightning.d.ts +1594 -0
  22. package/dist/src/lightning/lightning.d.ts.map +1 -0
  23. package/dist/src/lightning/lightning.js +468 -0
  24. package/dist/src/lightning/lightning.js.map +1 -0
  25. package/dist/src/mads/mads.d.ts +410 -0
  26. package/dist/src/mads/mads.d.ts.map +1 -0
  27. package/dist/src/mads/mads.js +219 -0
  28. package/dist/src/mads/mads.js.map +1 -0
  29. package/dist/src/mapie/mapie.d.ts +3680 -0
  30. package/dist/src/mapie/mapie.d.ts.map +1 -0
  31. package/dist/src/mapie/mapie.js +616 -0
  32. package/dist/src/mapie/mapie.js.map +1 -0
  33. package/dist/src/ngboost/ngboost.d.ts +483 -0
  34. package/dist/src/ngboost/ngboost.d.ts.map +1 -0
  35. package/dist/src/ngboost/ngboost.js +231 -0
  36. package/dist/src/ngboost/ngboost.js.map +1 -0
  37. package/dist/src/optimization/optimization.d.ts +700 -0
  38. package/dist/src/optimization/optimization.d.ts.map +1 -0
  39. package/dist/src/optimization/optimization.js +411 -0
  40. package/dist/src/optimization/optimization.js.map +1 -0
  41. package/dist/src/optuna/optuna.d.ts +857 -0
  42. package/dist/src/optuna/optuna.d.ts.map +1 -0
  43. package/dist/src/optuna/optuna.js +270 -0
  44. package/dist/src/optuna/optuna.js.map +1 -0
  45. package/dist/src/pymc/pymc.d.ts +2932 -0
  46. package/dist/src/pymc/pymc.d.ts.map +1 -0
  47. package/dist/src/pymc/pymc.js +688 -0
  48. package/dist/src/pymc/pymc.js.map +1 -0
  49. package/dist/src/scipy/scipy.d.ts +2205 -0
  50. package/dist/src/scipy/scipy.d.ts.map +1 -0
  51. package/dist/src/scipy/scipy.js +884 -0
  52. package/dist/src/scipy/scipy.js.map +1 -0
  53. package/dist/src/shap/shap.d.ts +4611 -0
  54. package/dist/src/shap/shap.d.ts.map +1 -0
  55. package/dist/src/shap/shap.js +552 -0
  56. package/dist/src/shap/shap.js.map +1 -0
  57. package/dist/src/simanneal/simanneal.d.ts +628 -0
  58. package/dist/src/simanneal/simanneal.d.ts.map +1 -0
  59. package/dist/src/simanneal/simanneal.js +328 -0
  60. package/dist/src/simanneal/simanneal.js.map +1 -0
  61. package/dist/src/simulation/simulation.d.ts +224 -0
  62. package/dist/src/simulation/simulation.d.ts.map +1 -0
  63. package/dist/src/simulation/simulation.js +189 -0
  64. package/dist/src/simulation/simulation.js.map +1 -0
  65. package/dist/src/sklearn/sklearn.d.ts +6362 -0
  66. package/dist/src/sklearn/sklearn.d.ts.map +1 -0
  67. package/dist/src/sklearn/sklearn.js +1508 -0
  68. package/dist/src/sklearn/sklearn.js.map +1 -0
  69. package/dist/src/torch/torch.d.ts +1205 -0
  70. package/dist/src/torch/torch.d.ts.map +1 -0
  71. package/dist/src/torch/torch.js +440 -0
  72. package/dist/src/torch/torch.js.map +1 -0
  73. package/dist/src/types.d.ts +43 -0
  74. package/dist/src/types.d.ts.map +1 -0
  75. package/dist/src/types.js +44 -0
  76. package/dist/src/types.js.map +1 -0
  77. package/dist/src/xgboost/xgboost.d.ts +1424 -0
  78. package/dist/src/xgboost/xgboost.d.ts.map +1 -0
  79. package/dist/src/xgboost/xgboost.js +432 -0
  80. package/dist/src/xgboost/xgboost.js.map +1 -0
  81. package/package.json +3 -3
@@ -0,0 +1,483 @@
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
+ * NGBoost platform functions for East.
7
+ *
8
+ * Provides probabilistic predictions with natural gradient boosting.
9
+ * Returns mean, standard deviation, and confidence intervals.
10
+ * Uses cloudpickle for model serialization.
11
+ *
12
+ * @packageDocumentation
13
+ */
14
+ import { StructType, VariantType, OptionType, IntegerType, FloatType, BlobType, NullType } from "@elaraai/east";
15
+ import { VectorType, MatrixType } from "../types.js";
16
+ export { VectorType, MatrixType } from "../types.js";
17
+ /**
18
+ * Distribution type for NGBoost.
19
+ */
20
+ export declare const NGBoostDistributionType: VariantType<{
21
+ /** Normal (Gaussian) distribution */
22
+ readonly normal: NullType;
23
+ /** Log-normal distribution (for positive targets) */
24
+ readonly lognormal: NullType;
25
+ }>;
26
+ /**
27
+ * Configuration for NGBoost models.
28
+ */
29
+ export declare const NGBoostConfigType: StructType<{
30
+ /** Number of boosting rounds (default 500) */
31
+ readonly n_estimators: OptionType<IntegerType>;
32
+ /** Learning rate / step size shrinkage (default 0.01) */
33
+ readonly learning_rate: OptionType<FloatType>;
34
+ /** Fraction of samples to use in each iteration (default 1.0) */
35
+ readonly minibatch_frac: OptionType<FloatType>;
36
+ /** Fraction of features to use in each iteration (default 1.0) */
37
+ readonly col_sample: OptionType<FloatType>;
38
+ /** Random seed for reproducibility */
39
+ readonly random_state: OptionType<IntegerType>;
40
+ /** Distribution type (default normal) */
41
+ readonly distribution: OptionType<VariantType<{
42
+ /** Normal (Gaussian) distribution */
43
+ readonly normal: NullType;
44
+ /** Log-normal distribution (for positive targets) */
45
+ readonly lognormal: NullType;
46
+ }>>;
47
+ }>;
48
+ /**
49
+ * Configuration for NGBoost predictions with uncertainty.
50
+ */
51
+ export declare const NGBoostPredictConfigType: StructType<{
52
+ /** Confidence level for intervals (default 0.95) */
53
+ readonly confidence_level: OptionType<FloatType>;
54
+ }>;
55
+ /**
56
+ * Result type for probabilistic predictions.
57
+ */
58
+ export declare const NGBoostPredictResultType: StructType<{
59
+ /** Point predictions (mean) */
60
+ readonly predictions: VectorType<FloatType>;
61
+ /** Standard deviation */
62
+ readonly std: OptionType<VectorType<FloatType>>;
63
+ /** Lower confidence interval */
64
+ readonly lower: OptionType<VectorType<FloatType>>;
65
+ /** Upper confidence interval */
66
+ readonly upper: OptionType<VectorType<FloatType>>;
67
+ }>;
68
+ /**
69
+ * Model blob type for serialized NGBoost models.
70
+ */
71
+ export declare const NGBoostModelBlobType: VariantType<{
72
+ /** NGBoost regressor model */
73
+ readonly ngboost_regressor: StructType<{
74
+ /** Cloudpickle serialized model */
75
+ readonly data: BlobType;
76
+ /** Distribution type used */
77
+ readonly distribution: VariantType<{
78
+ /** Normal (Gaussian) distribution */
79
+ readonly normal: NullType;
80
+ /** Log-normal distribution (for positive targets) */
81
+ readonly lognormal: NullType;
82
+ }>;
83
+ /** Number of input features */
84
+ readonly n_features: IntegerType;
85
+ }>;
86
+ }>;
87
+ /**
88
+ * Train an NGBoost regression model with probabilistic output.
89
+ *
90
+ * @param X - Feature matrix
91
+ * @param y - Target vector
92
+ * @param config - NGBoost configuration
93
+ * @returns Model blob containing trained regressor
94
+ */
95
+ export declare const ngboost_train_regressor: import("@elaraai/east").PlatformDefinition<[MatrixType<FloatType>, VectorType<FloatType>, StructType<{
96
+ /** Number of boosting rounds (default 500) */
97
+ readonly n_estimators: OptionType<IntegerType>;
98
+ /** Learning rate / step size shrinkage (default 0.01) */
99
+ readonly learning_rate: OptionType<FloatType>;
100
+ /** Fraction of samples to use in each iteration (default 1.0) */
101
+ readonly minibatch_frac: OptionType<FloatType>;
102
+ /** Fraction of features to use in each iteration (default 1.0) */
103
+ readonly col_sample: OptionType<FloatType>;
104
+ /** Random seed for reproducibility */
105
+ readonly random_state: OptionType<IntegerType>;
106
+ /** Distribution type (default normal) */
107
+ readonly distribution: OptionType<VariantType<{
108
+ /** Normal (Gaussian) distribution */
109
+ readonly normal: NullType;
110
+ /** Log-normal distribution (for positive targets) */
111
+ readonly lognormal: NullType;
112
+ }>>;
113
+ }>], VariantType<{
114
+ /** NGBoost regressor model */
115
+ readonly ngboost_regressor: StructType<{
116
+ /** Cloudpickle serialized model */
117
+ readonly data: BlobType;
118
+ /** Distribution type used */
119
+ readonly distribution: VariantType<{
120
+ /** Normal (Gaussian) distribution */
121
+ readonly normal: NullType;
122
+ /** Log-normal distribution (for positive targets) */
123
+ readonly lognormal: NullType;
124
+ }>;
125
+ /** Number of input features */
126
+ readonly n_features: IntegerType;
127
+ }>;
128
+ }>>;
129
+ /**
130
+ * Make point predictions (mean) with a trained NGBoost regressor.
131
+ *
132
+ * @param model - Trained regressor model blob
133
+ * @param X - Feature matrix
134
+ * @returns Predicted values (mean of predictive distribution)
135
+ */
136
+ export declare const ngboost_predict: import("@elaraai/east").PlatformDefinition<[VariantType<{
137
+ /** NGBoost regressor model */
138
+ readonly ngboost_regressor: StructType<{
139
+ /** Cloudpickle serialized model */
140
+ readonly data: BlobType;
141
+ /** Distribution type used */
142
+ readonly distribution: VariantType<{
143
+ /** Normal (Gaussian) distribution */
144
+ readonly normal: NullType;
145
+ /** Log-normal distribution (for positive targets) */
146
+ readonly lognormal: NullType;
147
+ }>;
148
+ /** Number of input features */
149
+ readonly n_features: IntegerType;
150
+ }>;
151
+ }>, MatrixType<FloatType>], VectorType<FloatType>>;
152
+ /**
153
+ * Get predictions with full uncertainty from NGBoost regressor.
154
+ *
155
+ * Returns mean, standard deviation, and confidence intervals.
156
+ *
157
+ * @param model - Trained regressor model blob
158
+ * @param X - Feature matrix
159
+ * @param config - Prediction configuration (confidence level)
160
+ * @returns Predictions with uncertainty estimates
161
+ */
162
+ export declare const ngboost_predict_dist: import("@elaraai/east").PlatformDefinition<[VariantType<{
163
+ /** NGBoost regressor model */
164
+ readonly ngboost_regressor: StructType<{
165
+ /** Cloudpickle serialized model */
166
+ readonly data: BlobType;
167
+ /** Distribution type used */
168
+ readonly distribution: VariantType<{
169
+ /** Normal (Gaussian) distribution */
170
+ readonly normal: NullType;
171
+ /** Log-normal distribution (for positive targets) */
172
+ readonly lognormal: NullType;
173
+ }>;
174
+ /** Number of input features */
175
+ readonly n_features: IntegerType;
176
+ }>;
177
+ }>, MatrixType<FloatType>, StructType<{
178
+ /** Confidence level for intervals (default 0.95) */
179
+ readonly confidence_level: OptionType<FloatType>;
180
+ }>], StructType<{
181
+ /** Point predictions (mean) */
182
+ readonly predictions: VectorType<FloatType>;
183
+ /** Standard deviation */
184
+ readonly std: OptionType<VectorType<FloatType>>;
185
+ /** Lower confidence interval */
186
+ readonly lower: OptionType<VectorType<FloatType>>;
187
+ /** Upper confidence interval */
188
+ readonly upper: OptionType<VectorType<FloatType>>;
189
+ }>>;
190
+ /**
191
+ * Type definitions for NGBoost functions.
192
+ */
193
+ export declare const NGBoostTypes: {
194
+ /** Distribution type for NGBoost */
195
+ readonly NGBoostDistributionType: VariantType<{
196
+ /** Normal (Gaussian) distribution */
197
+ readonly normal: NullType;
198
+ /** Log-normal distribution (for positive targets) */
199
+ readonly lognormal: NullType;
200
+ }>;
201
+ /** NGBoost configuration type */
202
+ readonly NGBoostConfigType: StructType<{
203
+ /** Number of boosting rounds (default 500) */
204
+ readonly n_estimators: OptionType<IntegerType>;
205
+ /** Learning rate / step size shrinkage (default 0.01) */
206
+ readonly learning_rate: OptionType<FloatType>;
207
+ /** Fraction of samples to use in each iteration (default 1.0) */
208
+ readonly minibatch_frac: OptionType<FloatType>;
209
+ /** Fraction of features to use in each iteration (default 1.0) */
210
+ readonly col_sample: OptionType<FloatType>;
211
+ /** Random seed for reproducibility */
212
+ readonly random_state: OptionType<IntegerType>;
213
+ /** Distribution type (default normal) */
214
+ readonly distribution: OptionType<VariantType<{
215
+ /** Normal (Gaussian) distribution */
216
+ readonly normal: NullType;
217
+ /** Log-normal distribution (for positive targets) */
218
+ readonly lognormal: NullType;
219
+ }>>;
220
+ }>;
221
+ /** Prediction configuration type */
222
+ readonly NGBoostPredictConfigType: StructType<{
223
+ /** Confidence level for intervals (default 0.95) */
224
+ readonly confidence_level: OptionType<FloatType>;
225
+ }>;
226
+ /** Prediction result type */
227
+ readonly NGBoostPredictResultType: StructType<{
228
+ /** Point predictions (mean) */
229
+ readonly predictions: VectorType<FloatType>;
230
+ /** Standard deviation */
231
+ readonly std: OptionType<VectorType<FloatType>>;
232
+ /** Lower confidence interval */
233
+ readonly lower: OptionType<VectorType<FloatType>>;
234
+ /** Upper confidence interval */
235
+ readonly upper: OptionType<VectorType<FloatType>>;
236
+ }>;
237
+ /** Model blob type for NGBoost models */
238
+ readonly ModelBlobType: VariantType<{
239
+ /** NGBoost regressor model */
240
+ readonly ngboost_regressor: StructType<{
241
+ /** Cloudpickle serialized model */
242
+ readonly data: BlobType;
243
+ /** Distribution type used */
244
+ readonly distribution: VariantType<{
245
+ /** Normal (Gaussian) distribution */
246
+ readonly normal: NullType;
247
+ /** Log-normal distribution (for positive targets) */
248
+ readonly lognormal: NullType;
249
+ }>;
250
+ /** Number of input features */
251
+ readonly n_features: IntegerType;
252
+ }>;
253
+ }>;
254
+ };
255
+ /**
256
+ * NGBoost probabilistic gradient boosting.
257
+ *
258
+ * Provides regression with uncertainty quantification using natural gradient boosting.
259
+ *
260
+ * @example
261
+ * ```ts
262
+ * import { East, variant } from "@elaraai/east";
263
+ * import { NGBoost } from "@elaraai/east-py-datascience";
264
+ *
265
+ * const train = East.function([], NGBoost.Types.ModelBlobType, $ => {
266
+ * const X = $.let([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0], [7.0, 8.0]]);
267
+ * const y = $.let([1.0, 2.0, 3.0, 4.0]);
268
+ * const config = $.let({
269
+ * n_estimators: variant('some', 100n),
270
+ * learning_rate: variant('some', 0.01),
271
+ * minibatch_frac: variant('none', null),
272
+ * col_sample: variant('none', null),
273
+ * random_state: variant('some', 42n),
274
+ * distribution: variant('none', null),
275
+ * });
276
+ * return $.return(NGBoost.trainRegressor(X, y, config));
277
+ * });
278
+ * ```
279
+ */
280
+ export declare const NGBoost: {
281
+ /**
282
+ * Train an NGBoost regression model with probabilistic output.
283
+ *
284
+ * @example
285
+ * ```ts
286
+ * import { East, FloatType, MatrixType, VectorType, variant } from "@elaraai/east";
287
+ * import { NGBoost, NGBoostConfigType } from "@elaraai/east-py-datascience";
288
+ *
289
+ * const train = East.function([], NGBoost.Types.ModelBlobType, ($) => {
290
+ * const X = $.let([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0], [7.0, 8.0]]);
291
+ * const y = $.let(new Float64Array([1.0, 2.0, 3.0, 4.0]));
292
+ * const config = $.let({
293
+ * n_estimators: variant("some", 100n), learning_rate: variant("some", 0.01),
294
+ * minibatch_frac: variant("none", null), col_sample: variant("none", null),
295
+ * random_state: variant("some", 42n), distribution: variant("none", null),
296
+ * }, NGBoostConfigType);
297
+ * return $.return(NGBoost.trainRegressor(X, y, config));
298
+ * });
299
+ * ```
300
+ */
301
+ readonly trainRegressor: import("@elaraai/east").PlatformDefinition<[MatrixType<FloatType>, VectorType<FloatType>, StructType<{
302
+ /** Number of boosting rounds (default 500) */
303
+ readonly n_estimators: OptionType<IntegerType>;
304
+ /** Learning rate / step size shrinkage (default 0.01) */
305
+ readonly learning_rate: OptionType<FloatType>;
306
+ /** Fraction of samples to use in each iteration (default 1.0) */
307
+ readonly minibatch_frac: OptionType<FloatType>;
308
+ /** Fraction of features to use in each iteration (default 1.0) */
309
+ readonly col_sample: OptionType<FloatType>;
310
+ /** Random seed for reproducibility */
311
+ readonly random_state: OptionType<IntegerType>;
312
+ /** Distribution type (default normal) */
313
+ readonly distribution: OptionType<VariantType<{
314
+ /** Normal (Gaussian) distribution */
315
+ readonly normal: NullType;
316
+ /** Log-normal distribution (for positive targets) */
317
+ readonly lognormal: NullType;
318
+ }>>;
319
+ }>], VariantType<{
320
+ /** NGBoost regressor model */
321
+ readonly ngboost_regressor: StructType<{
322
+ /** Cloudpickle serialized model */
323
+ readonly data: BlobType;
324
+ /** Distribution type used */
325
+ readonly distribution: VariantType<{
326
+ /** Normal (Gaussian) distribution */
327
+ readonly normal: NullType;
328
+ /** Log-normal distribution (for positive targets) */
329
+ readonly lognormal: NullType;
330
+ }>;
331
+ /** Number of input features */
332
+ readonly n_features: IntegerType;
333
+ }>;
334
+ }>>;
335
+ /**
336
+ * Make point predictions with a trained NGBoost regressor.
337
+ *
338
+ * @example
339
+ * ```ts
340
+ * import { East, FloatType, MatrixType, VectorType } from "@elaraai/east";
341
+ * import { NGBoost } from "@elaraai/east-py-datascience";
342
+ *
343
+ * const predictFn = East.function(
344
+ * [NGBoost.Types.ModelBlobType, MatrixType(FloatType)],
345
+ * VectorType(FloatType),
346
+ * ($, model, X) => {
347
+ * return $.return(NGBoost.predict(model, X));
348
+ * }
349
+ * );
350
+ * ```
351
+ */
352
+ readonly predict: import("@elaraai/east").PlatformDefinition<[VariantType<{
353
+ /** NGBoost regressor model */
354
+ readonly ngboost_regressor: StructType<{
355
+ /** Cloudpickle serialized model */
356
+ readonly data: BlobType;
357
+ /** Distribution type used */
358
+ readonly distribution: VariantType<{
359
+ /** Normal (Gaussian) distribution */
360
+ readonly normal: NullType;
361
+ /** Log-normal distribution (for positive targets) */
362
+ readonly lognormal: NullType;
363
+ }>;
364
+ /** Number of input features */
365
+ readonly n_features: IntegerType;
366
+ }>;
367
+ }>, MatrixType<FloatType>], VectorType<FloatType>>;
368
+ /**
369
+ * Get predictions with uncertainty estimates (confidence intervals, std).
370
+ *
371
+ * @example
372
+ * ```ts
373
+ * import { East, FloatType, MatrixType, variant } from "@elaraai/east";
374
+ * import { NGBoost, NGBoostPredictConfigType } from "@elaraai/east-py-datascience";
375
+ *
376
+ * const predictFn = East.function(
377
+ * [NGBoost.Types.ModelBlobType, MatrixType(FloatType)],
378
+ * NGBoost.Types.NGBoostPredictResultType,
379
+ * ($, model, X) => {
380
+ * const config = $.let({
381
+ * confidence_level: variant("some", 0.95),
382
+ * }, NGBoostPredictConfigType);
383
+ * const result = $.let(NGBoost.predictDist(model, X, config));
384
+ * // result.predictions => point predictions
385
+ * // result.lower, result.upper => confidence interval
386
+ * return $.return(result);
387
+ * }
388
+ * );
389
+ * ```
390
+ */
391
+ readonly predictDist: import("@elaraai/east").PlatformDefinition<[VariantType<{
392
+ /** NGBoost regressor model */
393
+ readonly ngboost_regressor: StructType<{
394
+ /** Cloudpickle serialized model */
395
+ readonly data: BlobType;
396
+ /** Distribution type used */
397
+ readonly distribution: VariantType<{
398
+ /** Normal (Gaussian) distribution */
399
+ readonly normal: NullType;
400
+ /** Log-normal distribution (for positive targets) */
401
+ readonly lognormal: NullType;
402
+ }>;
403
+ /** Number of input features */
404
+ readonly n_features: IntegerType;
405
+ }>;
406
+ }>, MatrixType<FloatType>, StructType<{
407
+ /** Confidence level for intervals (default 0.95) */
408
+ readonly confidence_level: OptionType<FloatType>;
409
+ }>], StructType<{
410
+ /** Point predictions (mean) */
411
+ readonly predictions: VectorType<FloatType>;
412
+ /** Standard deviation */
413
+ readonly std: OptionType<VectorType<FloatType>>;
414
+ /** Lower confidence interval */
415
+ readonly lower: OptionType<VectorType<FloatType>>;
416
+ /** Upper confidence interval */
417
+ readonly upper: OptionType<VectorType<FloatType>>;
418
+ }>>;
419
+ /** Type definitions */
420
+ readonly Types: {
421
+ /** Distribution type for NGBoost */
422
+ readonly NGBoostDistributionType: VariantType<{
423
+ /** Normal (Gaussian) distribution */
424
+ readonly normal: NullType;
425
+ /** Log-normal distribution (for positive targets) */
426
+ readonly lognormal: NullType;
427
+ }>;
428
+ /** NGBoost configuration type */
429
+ readonly NGBoostConfigType: StructType<{
430
+ /** Number of boosting rounds (default 500) */
431
+ readonly n_estimators: OptionType<IntegerType>;
432
+ /** Learning rate / step size shrinkage (default 0.01) */
433
+ readonly learning_rate: OptionType<FloatType>;
434
+ /** Fraction of samples to use in each iteration (default 1.0) */
435
+ readonly minibatch_frac: OptionType<FloatType>;
436
+ /** Fraction of features to use in each iteration (default 1.0) */
437
+ readonly col_sample: OptionType<FloatType>;
438
+ /** Random seed for reproducibility */
439
+ readonly random_state: OptionType<IntegerType>;
440
+ /** Distribution type (default normal) */
441
+ readonly distribution: OptionType<VariantType<{
442
+ /** Normal (Gaussian) distribution */
443
+ readonly normal: NullType;
444
+ /** Log-normal distribution (for positive targets) */
445
+ readonly lognormal: NullType;
446
+ }>>;
447
+ }>;
448
+ /** Prediction configuration type */
449
+ readonly NGBoostPredictConfigType: StructType<{
450
+ /** Confidence level for intervals (default 0.95) */
451
+ readonly confidence_level: OptionType<FloatType>;
452
+ }>;
453
+ /** Prediction result type */
454
+ readonly NGBoostPredictResultType: StructType<{
455
+ /** Point predictions (mean) */
456
+ readonly predictions: VectorType<FloatType>;
457
+ /** Standard deviation */
458
+ readonly std: OptionType<VectorType<FloatType>>;
459
+ /** Lower confidence interval */
460
+ readonly lower: OptionType<VectorType<FloatType>>;
461
+ /** Upper confidence interval */
462
+ readonly upper: OptionType<VectorType<FloatType>>;
463
+ }>;
464
+ /** Model blob type for NGBoost models */
465
+ readonly ModelBlobType: VariantType<{
466
+ /** NGBoost regressor model */
467
+ readonly ngboost_regressor: StructType<{
468
+ /** Cloudpickle serialized model */
469
+ readonly data: BlobType;
470
+ /** Distribution type used */
471
+ readonly distribution: VariantType<{
472
+ /** Normal (Gaussian) distribution */
473
+ readonly normal: NullType;
474
+ /** Log-normal distribution (for positive targets) */
475
+ readonly lognormal: NullType;
476
+ }>;
477
+ /** Number of input features */
478
+ readonly n_features: IntegerType;
479
+ }>;
480
+ }>;
481
+ };
482
+ };
483
+ //# sourceMappingURL=ngboost.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"ngboost.d.ts","sourceRoot":"","sources":["../../../src/ngboost/ngboost.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;;GAQG;AAEH,OAAO,EAEH,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,QAAQ,EACR,QAAQ,EACX,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGrD,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAMrD;;GAEG;AACH,eAAO,MAAM,uBAAuB;IAChC,qCAAqC;;IAErC,qDAAqD;;EAEvD,CAAC;AAMH;;GAEG;AACH,eAAO,MAAM,iBAAiB;IAC1B,8CAA8C;;IAE9C,yDAAyD;;IAEzD,iEAAiE;;IAEjE,kEAAkE;;IAElE,sCAAsC;;IAEtC,yCAAyC;;QAxBzC,qCAAqC;;QAErC,qDAAqD;;;EAwBvD,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,wBAAwB;IACjC,oDAAoD;;EAEtD,CAAC;AAMH;;GAEG;AACH,eAAO,MAAM,wBAAwB;IACjC,+BAA+B;;IAE/B,yBAAyB;;IAEzB,gCAAgC;;IAEhC,gCAAgC;;EAElC,CAAC;AAMH;;GAEG;AACH,eAAO,MAAM,oBAAoB;IAC7B,8BAA8B;;QAE1B,mCAAmC;;QAEnC,6BAA6B;;YAlEjC,qCAAqC;;YAErC,qDAAqD;;;QAkEjD,+BAA+B;;;EAGrC,CAAC;AAMH;;;;;;;GAOG;AACH,eAAO,MAAM,uBAAuB;IAvEhC,8CAA8C;;IAE9C,yDAAyD;;IAEzD,iEAAiE;;IAEjE,kEAAkE;;IAElE,sCAAsC;;IAEtC,yCAAyC;;QAxBzC,qCAAqC;;QAErC,qDAAqD;;;;IA4DrD,8BAA8B;;QAE1B,mCAAmC;;QAEnC,6BAA6B;;YAlEjC,qCAAqC;;YAErC,qDAAqD;;;QAkEjD,+BAA+B;;;GAqBtC,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,eAAe;IApCxB,8BAA8B;;QAE1B,mCAAmC;;QAEnC,6BAA6B;;YAlEjC,qCAAqC;;YAErC,qDAAqD;;;QAkEjD,+BAA+B;;;kDAkCtC,CAAC;AAEF;;;;;;;;;GASG;AACH,eAAO,MAAM,oBAAoB;IApD7B,8BAA8B;;QAE1B,mCAAmC;;QAEnC,6BAA6B;;YAlEjC,qCAAqC;;YAErC,qDAAqD;;;QAkEjD,+BAA+B;;;;IApCnC,oDAAoD;;;IAYpD,+BAA+B;;IAE/B,yBAAyB;;IAEzB,gCAAgC;;IAEhC,gCAAgC;;GAoEnC,CAAC;AAMF;;GAEG;AACH,eAAO,MAAM,YAAY;IACrB,oCAAoC;;QAhIpC,qCAAqC;;QAErC,qDAAqD;;;IAgIrD,iCAAiC;;QApHjC,8CAA8C;;QAE9C,yDAAyD;;QAEzD,iEAAiE;;QAEjE,kEAAkE;;QAElE,sCAAsC;;QAEtC,yCAAyC;;YAxBzC,qCAAqC;;YAErC,qDAAqD;;;;IAkIrD,oCAAoC;;QApGpC,oDAAoD;;;IAsGpD,6BAA6B;;QA1F7B,+BAA+B;;QAE/B,yBAAyB;;QAEzB,gCAAgC;;QAEhC,gCAAgC;;;IAsFhC,yCAAyC;;QA1EzC,8BAA8B;;YAE1B,mCAAmC;;YAEnC,6BAA6B;;gBAlEjC,qCAAqC;;gBAErC,qDAAqD;;;YAkEjD,+BAA+B;;;;CAsE7B,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,eAAO,MAAM,OAAO;IAChB;;;;;;;;;;;;;;;;;;;OAmBG;;QA3KH,8CAA8C;;QAE9C,yDAAyD;;QAEzD,iEAAiE;;QAEjE,kEAAkE;;QAElE,sCAAsC;;QAEtC,yCAAyC;;YAxBzC,qCAAqC;;YAErC,qDAAqD;;;;QA4DrD,8BAA8B;;YAE1B,mCAAmC;;YAEnC,6BAA6B;;gBAlEjC,qCAAqC;;gBAErC,qDAAqD;;;YAkEjD,+BAA+B;;;;IAwHnC;;;;;;;;;;;;;;;;OAgBG;;QA9IH,8BAA8B;;YAE1B,mCAAmC;;YAEnC,6BAA6B;;gBAlEjC,qCAAqC;;gBAErC,qDAAqD;;;YAkEjD,+BAA+B;;;;IA2InC;;;;;;;;;;;;;;;;;;;;;;OAsBG;;QAvKH,8BAA8B;;YAE1B,mCAAmC;;YAEnC,6BAA6B;;gBAlEjC,qCAAqC;;gBAErC,qDAAqD;;;YAkEjD,+BAA+B;;;;QApCnC,oDAAoD;;;QAYpD,+BAA+B;;QAE/B,yBAAyB;;QAEzB,gCAAgC;;QAEhC,gCAAgC;;;IAqLhC,uBAAuB;;QAvGvB,oCAAoC;;YAhIpC,qCAAqC;;YAErC,qDAAqD;;;QAgIrD,iCAAiC;;YApHjC,8CAA8C;;YAE9C,yDAAyD;;YAEzD,iEAAiE;;YAEjE,kEAAkE;;YAElE,sCAAsC;;YAEtC,yCAAyC;;gBAxBzC,qCAAqC;;gBAErC,qDAAqD;;;;QAkIrD,oCAAoC;;YApGpC,oDAAoD;;;QAsGpD,6BAA6B;;YA1F7B,+BAA+B;;YAE/B,yBAAyB;;YAEzB,gCAAgC;;YAEhC,gCAAgC;;;QAsFhC,yCAAyC;;YA1EzC,8BAA8B;;gBAE1B,mCAAmC;;gBAEnC,6BAA6B;;oBAlEjC,qCAAqC;;oBAErC,qDAAqD;;;gBAkEjD,+BAA+B;;;;;CAqK7B,CAAC"}