@elaraai/east-py-datascience 1.0.0 → 1.0.1
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.
- package/package.json +3 -3
- package/dist/src/alns/alns.d.ts +0 -528
- package/dist/src/alns/alns.d.ts.map +0 -1
- package/dist/src/alns/alns.js +0 -238
- package/dist/src/alns/alns.js.map +0 -1
- package/dist/src/google_or/google_or.d.ts +0 -2457
- package/dist/src/google_or/google_or.d.ts.map +0 -1
- package/dist/src/google_or/google_or.js +0 -547
- package/dist/src/google_or/google_or.js.map +0 -1
- package/dist/src/gp/gp.d.ts +0 -447
- package/dist/src/gp/gp.d.ts.map +0 -1
- package/dist/src/gp/gp.js +0 -222
- package/dist/src/gp/gp.js.map +0 -1
- package/dist/src/index.d.ts +0 -34
- package/dist/src/index.d.ts.map +0 -1
- package/dist/src/index.js +0 -57
- package/dist/src/index.js.map +0 -1
- package/dist/src/lightgbm/lightgbm.d.ts +0 -575
- package/dist/src/lightgbm/lightgbm.d.ts.map +0 -1
- package/dist/src/lightgbm/lightgbm.js +0 -241
- package/dist/src/lightgbm/lightgbm.js.map +0 -1
- package/dist/src/lightning/lightning.d.ts +0 -1594
- package/dist/src/lightning/lightning.d.ts.map +0 -1
- package/dist/src/lightning/lightning.js +0 -468
- package/dist/src/lightning/lightning.js.map +0 -1
- package/dist/src/mads/mads.d.ts +0 -410
- package/dist/src/mads/mads.d.ts.map +0 -1
- package/dist/src/mads/mads.js +0 -219
- package/dist/src/mads/mads.js.map +0 -1
- package/dist/src/mapie/mapie.d.ts +0 -3680
- package/dist/src/mapie/mapie.d.ts.map +0 -1
- package/dist/src/mapie/mapie.js +0 -616
- package/dist/src/mapie/mapie.js.map +0 -1
- package/dist/src/ngboost/ngboost.d.ts +0 -483
- package/dist/src/ngboost/ngboost.d.ts.map +0 -1
- package/dist/src/ngboost/ngboost.js +0 -231
- package/dist/src/ngboost/ngboost.js.map +0 -1
- package/dist/src/optimization/optimization.d.ts +0 -700
- package/dist/src/optimization/optimization.d.ts.map +0 -1
- package/dist/src/optimization/optimization.js +0 -411
- package/dist/src/optimization/optimization.js.map +0 -1
- package/dist/src/optuna/optuna.d.ts +0 -857
- package/dist/src/optuna/optuna.d.ts.map +0 -1
- package/dist/src/optuna/optuna.js +0 -270
- package/dist/src/optuna/optuna.js.map +0 -1
- package/dist/src/pymc/pymc.d.ts +0 -2932
- package/dist/src/pymc/pymc.d.ts.map +0 -1
- package/dist/src/pymc/pymc.js +0 -688
- package/dist/src/pymc/pymc.js.map +0 -1
- package/dist/src/scipy/scipy.d.ts +0 -2205
- package/dist/src/scipy/scipy.d.ts.map +0 -1
- package/dist/src/scipy/scipy.js +0 -884
- package/dist/src/scipy/scipy.js.map +0 -1
- package/dist/src/shap/shap.d.ts +0 -4611
- package/dist/src/shap/shap.d.ts.map +0 -1
- package/dist/src/shap/shap.js +0 -552
- package/dist/src/shap/shap.js.map +0 -1
- package/dist/src/simanneal/simanneal.d.ts +0 -628
- package/dist/src/simanneal/simanneal.d.ts.map +0 -1
- package/dist/src/simanneal/simanneal.js +0 -328
- package/dist/src/simanneal/simanneal.js.map +0 -1
- package/dist/src/simulation/simulation.d.ts +0 -224
- package/dist/src/simulation/simulation.d.ts.map +0 -1
- package/dist/src/simulation/simulation.js +0 -189
- package/dist/src/simulation/simulation.js.map +0 -1
- package/dist/src/sklearn/sklearn.d.ts +0 -6362
- package/dist/src/sklearn/sklearn.d.ts.map +0 -1
- package/dist/src/sklearn/sklearn.js +0 -1508
- package/dist/src/sklearn/sklearn.js.map +0 -1
- package/dist/src/torch/torch.d.ts +0 -1205
- package/dist/src/torch/torch.d.ts.map +0 -1
- package/dist/src/torch/torch.js +0 -440
- package/dist/src/torch/torch.js.map +0 -1
- package/dist/src/types.d.ts +0 -43
- package/dist/src/types.d.ts.map +0 -1
- package/dist/src/types.js +0 -44
- package/dist/src/types.js.map +0 -1
- package/dist/src/xgboost/xgboost.d.ts +0 -1424
- package/dist/src/xgboost/xgboost.d.ts.map +0 -1
- package/dist/src/xgboost/xgboost.js +0 -432
- package/dist/src/xgboost/xgboost.js.map +0 -1
|
@@ -1,483 +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
|
-
* 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
|
|
@@ -1 +0,0 @@
|
|
|
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"}
|