@elaraai/east-py-datascience 0.0.2-beta.6 → 0.0.2-beta.61
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/dist/alns/alns.d.ts +528 -0
- package/dist/alns/alns.d.ts.map +1 -0
- package/dist/alns/alns.js +238 -0
- package/dist/alns/alns.js.map +1 -0
- package/dist/gp/gp.d.ts +120 -120
- package/dist/gp/gp.d.ts.map +1 -1
- package/dist/gp/gp.js +7 -7
- package/dist/gp/gp.js.map +1 -1
- package/dist/index.d.ts +5 -2
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +8 -2
- package/dist/index.js.map +1 -1
- package/dist/lightgbm/lightgbm.d.ts +168 -168
- package/dist/lightning/lightning.d.ts +1501 -0
- package/dist/lightning/lightning.d.ts.map +1 -0
- package/dist/lightning/lightning.js +376 -0
- package/dist/lightning/lightning.js.map +1 -0
- package/dist/mads/mads.d.ts +103 -103
- package/dist/mads/mads.d.ts.map +1 -1
- package/dist/mads/mads.js +5 -5
- package/dist/mads/mads.js.map +1 -1
- package/dist/mapie/mapie.d.ts +2058 -0
- package/dist/mapie/mapie.d.ts.map +1 -0
- package/dist/mapie/mapie.js +411 -0
- package/dist/mapie/mapie.js.map +1 -0
- package/dist/ngboost/ngboost.d.ts +126 -126
- package/dist/ngboost/ngboost.d.ts.map +1 -1
- package/dist/ngboost/ngboost.js +3 -3
- package/dist/ngboost/ngboost.js.map +1 -1
- package/dist/optuna/optuna.d.ts +314 -314
- package/dist/scipy/scipy.d.ts +535 -429
- package/dist/scipy/scipy.d.ts.map +1 -1
- package/dist/scipy/scipy.js +56 -3
- package/dist/scipy/scipy.js.map +1 -1
- package/dist/shap/shap.d.ts +1152 -358
- package/dist/shap/shap.d.ts.map +1 -1
- package/dist/shap/shap.js +189 -16
- package/dist/shap/shap.js.map +1 -1
- package/dist/simanneal/simanneal.d.ts +148 -148
- package/dist/sklearn/sklearn.d.ts +3104 -1316
- package/dist/sklearn/sklearn.d.ts.map +1 -1
- package/dist/sklearn/sklearn.js +325 -64
- package/dist/sklearn/sklearn.js.map +1 -1
- package/dist/torch/torch.d.ts +503 -350
- package/dist/torch/torch.d.ts.map +1 -1
- package/dist/torch/torch.js +39 -17
- package/dist/torch/torch.js.map +1 -1
- package/dist/tsconfig.tsbuildinfo +1 -1
- package/dist/xgboost/xgboost.d.ts +803 -178
- package/dist/xgboost/xgboost.d.ts.map +1 -1
- package/dist/xgboost/xgboost.js +102 -1
- package/dist/xgboost/xgboost.js.map +1 -1
- package/package.json +4 -4
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"xgboost.d.ts","sourceRoot":"","sources":["../../src/xgboost/xgboost.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EAEH,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,QAAQ,EACX,MAAM,eAAe,CAAC;AAIvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAMtE;;GAEG;AACH,eAAO,MAAM,iBAAiB;IAC1B,8CAA8C;;IAE9C,qCAAqC;;IAErC,wDAAwD;;IAExD,mEAAmE;;IAEnE,0DAA0D;;IAE1D,uEAAuE;;IAEvE,yCAAyC;;IAEzC,yCAAyC;;IAEzC,sCAAsC;;IAEtC,4DAA4D;;
|
|
1
|
+
{"version":3,"file":"xgboost.d.ts","sourceRoot":"","sources":["../../src/xgboost/xgboost.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EAEH,UAAU,EACV,WAAW,EACX,UAAU,EACV,SAAS,EACT,WAAW,EACX,SAAS,EACT,QAAQ,EACX,MAAM,eAAe,CAAC;AAIvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAMtE;;GAEG;AACH,eAAO,MAAM,iBAAiB;IAC1B,8CAA8C;;IAE9C,qCAAqC;;IAErC,wDAAwD;;IAExD,mEAAmE;;IAEnE,0DAA0D;;IAE1D,uEAAuE;;IAEvE,yCAAyC;;IAEzC,yCAAyC;;IAEzC,8EAA8E;;IAE9E,sCAAsC;;IAEtC,4DAA4D;;IAE5D,oEAAoE;;IAEpE,mEAAmE;;IAEnE,iHAAiH;;IAEjH,mFAAmF;;EAErF,CAAC;AAEH;;;GAGG;AACH,eAAO,MAAM,yBAAyB;IAClC,6EAA6E;;IAE7E,8CAA8C;;IAE9C,qCAAqC;;IAErC,wDAAwD;;IAExD,mEAAmE;;IAEnE,0DAA0D;;IAE1D,uEAAuE;;IAEvE,yCAAyC;;IAEzC,yCAAyC;;IAEzC,8EAA8E;;IAE9E,sCAAsC;;IAEtC,4DAA4D;;IAE5D,oEAAoE;;IAEpE,mEAAmE;;IAEnE,iHAAiH;;IAEjH,mFAAmF;;EAErF,CAAC;AAMH;;;;GAIG;AACH,eAAO,MAAM,oBAAoB;IAC7B,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;;IAGlE,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;;IAGlE,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;;EAGpE,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,gCAAgC;IACzC,0CAA0C;;IAE1C,oDAAoD;;EAEtD,CAAC;AAMH;;;;;;;GAOG;AACH,eAAO,MAAM,uBAAuB;IAxIhC,8CAA8C;;IAE9C,qCAAqC;;IAErC,wDAAwD;;IAExD,mEAAmE;;IAEnE,0DAA0D;;IAE1D,uEAAuE;;IAEvE,yCAAyC;;IAEzC,yCAAyC;;IAEzC,8EAA8E;;IAE9E,sCAAsC;;IAEtC,4DAA4D;;IAE5D,oEAAoE;;IAEpE,mEAAmE;;IAEnE,iHAAiH;;IAEjH,mFAAmF;;;IAqDnF,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;;IAGlE,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;;IAGlE,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;;GA+BrE,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,wBAAwB;IAtJjC,8CAA8C;;IAE9C,qCAAqC;;IAErC,wDAAwD;;IAExD,mEAAmE;;IAEnE,0DAA0D;;IAE1D,uEAAuE;;IAEvE,yCAAyC;;IAEzC,yCAAyC;;IAEzC,8EAA8E;;IAE9E,sCAAsC;;IAEtC,4DAA4D;;IAE5D,oEAAoE;;IAEpE,mEAAmE;;IAEnE,iHAAiH;;IAEjH,mFAAmF;;;IAqDnF,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;;IAGlE,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;;IAGlE,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;;GA6CrE,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,eAAe;IAlFxB,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;;IAGlE,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;;IAGlE,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;;2DA0DrE,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,qBAAqB;IA/F9B,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;;IAGlE,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;;IAGlE,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;;6DAuErE,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,qBAAqB;IA5G9B,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;;IAGlE,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;;IAGlE,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;;sEAoFrE,CAAC;AAEF;;;;;;;;;;GAUG;AACH,eAAO,MAAM,sBAAsB;IAzK/B,6EAA6E;;IAE7E,8CAA8C;;IAE9C,qCAAqC;;IAErC,wDAAwD;;IAExD,mEAAmE;;IAEnE,0DAA0D;;IAE1D,uEAAuE;;IAEvE,yCAAyC;;IAEzC,yCAAyC;;IAEzC,8EAA8E;;IAE9E,sCAAsC;;IAEtC,4DAA4D;;IAE5D,oEAAoE;;IAEpE,mEAAmE;;IAEnE,iHAAiH;;IAEjH,mFAAmF;;;IAcnF,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;;IAGlE,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;;IAGlE,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;;GAqGrE,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,wBAAwB;IA1IjC,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;;IAGlE,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;;IAGlE,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;;;IASlE,0CAA0C;;IAE1C,oDAAoD;;GAuGvD,CAAC;AAMF;;GAEG;AACH,eAAO,MAAM,YAAY;IACrB,oCAAoC;;IAEpC,uCAAuC;;IAEvC,4CAA4C;;IAE5C,iCAAiC;;QA/OjC,8CAA8C;;QAE9C,qCAAqC;;QAErC,wDAAwD;;QAExD,mEAAmE;;QAEnE,0DAA0D;;QAE1D,uEAAuE;;QAEvE,yCAAyC;;QAEzC,yCAAyC;;QAEzC,8EAA8E;;QAE9E,sCAAsC;;QAEtC,4DAA4D;;QAE5D,oEAAoE;;QAEpE,mEAAmE;;QAEnE,iHAAiH;;QAEjH,mFAAmF;;;IAqNnF,0CAA0C;;QA5M1C,6EAA6E;;QAE7E,8CAA8C;;QAE9C,qCAAqC;;QAErC,wDAAwD;;QAExD,mEAAmE;;QAEnE,0DAA0D;;QAE1D,uEAAuE;;QAEvE,yCAAyC;;QAEzC,yCAAyC;;QAEzC,8EAA8E;;QAE9E,sCAAsC;;QAEtC,4DAA4D;;QAE5D,oEAAoE;;QAEpE,mEAAmE;;QAEnE,iHAAiH;;QAEjH,mFAAmF;;;IAgLnF,sCAAsC;;QA7HtC,0CAA0C;;QAE1C,oDAAoD;;;IA6HpD,yCAAyC;;QApKzC,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;;QAGlE,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;;QAGlE,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;;;CA0I5D,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,eAAO,MAAM,OAAO;IAChB,8BAA8B;;QAxR9B,8CAA8C;;QAE9C,qCAAqC;;QAErC,wDAAwD;;QAExD,mEAAmE;;QAEnE,0DAA0D;;QAE1D,uEAAuE;;QAEvE,yCAAyC;;QAEzC,yCAAyC;;QAEzC,8EAA8E;;QAE9E,sCAAsC;;QAEtC,4DAA4D;;QAE5D,oEAAoE;;QAEpE,mEAAmE;;QAEnE,iHAAiH;;QAEjH,mFAAmF;;;QAqDnF,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;;QAGlE,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;;QAGlE,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;;;IA6KlE,+BAA+B;;QA1R/B,8CAA8C;;QAE9C,qCAAqC;;QAErC,wDAAwD;;QAExD,mEAAmE;;QAEnE,0DAA0D;;QAE1D,uEAAuE;;QAEvE,yCAAyC;;QAEzC,yCAAyC;;QAEzC,8EAA8E;;QAE9E,sCAAsC;;QAEtC,4DAA4D;;QAE5D,oEAAoE;;QAEpE,mEAAmE;;QAEnE,iHAAiH;;QAEjH,mFAAmF;;;QAqDnF,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;;QAGlE,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;;QAGlE,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;;;IA+KlE,uCAAuC;;QAvPvC,6EAA6E;;QAE7E,8CAA8C;;QAE9C,qCAAqC;;QAErC,wDAAwD;;QAExD,mEAAmE;;QAEnE,0DAA0D;;QAE1D,uEAAuE;;QAEvE,yCAAyC;;QAEzC,yCAAyC;;QAEzC,8EAA8E;;QAE9E,sCAAsC;;QAEtC,4DAA4D;;QAE5D,oEAAoE;;QAEpE,mEAAmE;;QAEnE,iHAAiH;;QAEjH,mFAAmF;;;QAcnF,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;;QAGlE,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;;QAGlE,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;;;IAiLlE,sCAAsC;;QA7MtC,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;;QAGlE,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;;QAGlE,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;;;IAmLlE,2CAA2C;;QA/M3C,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;;QAGlE,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;;QAGlE,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;;;IAqLlE,8CAA8C;;QAjN9C,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;;QAGlE,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;;QAGlE,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;;;IAuLlE,gDAAgD;;QAnNhD,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;;QAGlE,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;;QAGlE,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;;;QASlE,0CAA0C;;QAE1C,oDAAoD;;;IA8KpD,uBAAuB;;QA7DvB,oCAAoC;;QAEpC,uCAAuC;;QAEvC,4CAA4C;;QAE5C,iCAAiC;;YA/OjC,8CAA8C;;YAE9C,qCAAqC;;YAErC,wDAAwD;;YAExD,mEAAmE;;YAEnE,0DAA0D;;YAE1D,uEAAuE;;YAEvE,yCAAyC;;YAEzC,yCAAyC;;YAEzC,8EAA8E;;YAE9E,sCAAsC;;YAEtC,4DAA4D;;YAE5D,oEAAoE;;YAEpE,mEAAmE;;YAEnE,iHAAiH;;YAEjH,mFAAmF;;;QAqNnF,0CAA0C;;YA5M1C,6EAA6E;;YAE7E,8CAA8C;;YAE9C,qCAAqC;;YAErC,wDAAwD;;YAExD,mEAAmE;;YAEnE,0DAA0D;;YAE1D,uEAAuE;;YAEvE,yCAAyC;;YAEzC,yCAAyC;;YAEzC,8EAA8E;;YAE9E,sCAAsC;;YAEtC,4DAA4D;;YAE5D,oEAAoE;;YAEpE,mEAAmE;;YAEnE,iHAAiH;;YAEjH,mFAAmF;;;QAgLnF,sCAAsC;;YA7HtC,0CAA0C;;YAE1C,oDAAoD;;;QA6HpD,yCAAyC;;YApKzC,8BAA8B;;gBAE1B,mCAAmC;;gBAEnC,+BAA+B;;gBAE/B,8DAA8D;;;YAGlE,+BAA+B;;gBAE3B,mCAAmC;;gBAEnC,+BAA+B;;gBAE/B,wBAAwB;;gBAExB,8DAA8D;;;YAGlE,qEAAqE;;gBAEjE,uDAAuD;;gBAEvD,oCAAoC;;gBAEpC,+BAA+B;;gBAE/B,8DAA8D;;;;;CA2L5D,CAAC"}
|
package/dist/xgboost/xgboost.js
CHANGED
|
@@ -10,7 +10,7 @@
|
|
|
10
10
|
*
|
|
11
11
|
* @packageDocumentation
|
|
12
12
|
*/
|
|
13
|
-
import { East, StructType, VariantType, OptionType, IntegerType, FloatType, BlobType, } from "@elaraai/east";
|
|
13
|
+
import { East, StructType, VariantType, OptionType, ArrayType, IntegerType, FloatType, BlobType, } from "@elaraai/east";
|
|
14
14
|
import { VectorType, MatrixType, LabelVectorType } from "../types.js";
|
|
15
15
|
// Re-export shared types for convenience
|
|
16
16
|
export { VectorType, MatrixType, LabelVectorType } from "../types.js";
|
|
@@ -37,10 +37,58 @@ export const XGBoostConfigType = StructType({
|
|
|
37
37
|
reg_alpha: OptionType(FloatType),
|
|
38
38
|
/** L2 regularization term (default 1) */
|
|
39
39
|
reg_lambda: OptionType(FloatType),
|
|
40
|
+
/** Minimum loss reduction required to make a further partition (default 0) */
|
|
41
|
+
gamma: OptionType(FloatType),
|
|
40
42
|
/** Random seed for reproducibility */
|
|
41
43
|
random_state: OptionType(IntegerType),
|
|
42
44
|
/** Number of parallel threads (default -1 for all cores) */
|
|
43
45
|
n_jobs: OptionType(IntegerType),
|
|
46
|
+
/** Sample weights for training (one per sample, default uniform) */
|
|
47
|
+
sample_weight: OptionType(VectorType),
|
|
48
|
+
/** Column indices that contain categorical features (0-indexed) */
|
|
49
|
+
categorical_features: OptionType(ArrayType(IntegerType)),
|
|
50
|
+
/** Max categories for one-hot encoding (default 4). Features with more categories use partition-based splits. */
|
|
51
|
+
max_cat_to_onehot: OptionType(IntegerType),
|
|
52
|
+
/** Max categories considered per split for partition-based method (default 64). */
|
|
53
|
+
max_cat_threshold: OptionType(IntegerType),
|
|
54
|
+
});
|
|
55
|
+
/**
|
|
56
|
+
* Configuration for XGBoost quantile regression.
|
|
57
|
+
* Trains separate models for each quantile.
|
|
58
|
+
*/
|
|
59
|
+
export const XGBoostQuantileConfigType = StructType({
|
|
60
|
+
/** Quantiles to predict (e.g., [0.1, 0.5, 0.9] for 80% interval + median) */
|
|
61
|
+
quantiles: VectorType,
|
|
62
|
+
/** Number of boosting rounds (default 100) */
|
|
63
|
+
n_estimators: OptionType(IntegerType),
|
|
64
|
+
/** Maximum tree depth (default 6) */
|
|
65
|
+
max_depth: OptionType(IntegerType),
|
|
66
|
+
/** Learning rate / step size shrinkage (default 0.3) */
|
|
67
|
+
learning_rate: OptionType(FloatType),
|
|
68
|
+
/** Minimum sum of instance weight needed in a child (default 1) */
|
|
69
|
+
min_child_weight: OptionType(IntegerType),
|
|
70
|
+
/** Subsample ratio of training instances (default 1.0) */
|
|
71
|
+
subsample: OptionType(FloatType),
|
|
72
|
+
/** Subsample ratio of columns when constructing trees (default 1.0) */
|
|
73
|
+
colsample_bytree: OptionType(FloatType),
|
|
74
|
+
/** L1 regularization term (default 0) */
|
|
75
|
+
reg_alpha: OptionType(FloatType),
|
|
76
|
+
/** L2 regularization term (default 1) */
|
|
77
|
+
reg_lambda: OptionType(FloatType),
|
|
78
|
+
/** Minimum loss reduction required to make a further partition (default 0) */
|
|
79
|
+
gamma: OptionType(FloatType),
|
|
80
|
+
/** Random seed for reproducibility */
|
|
81
|
+
random_state: OptionType(IntegerType),
|
|
82
|
+
/** Number of parallel threads (default -1 for all cores) */
|
|
83
|
+
n_jobs: OptionType(IntegerType),
|
|
84
|
+
/** Sample weights for training (one per sample, default uniform) */
|
|
85
|
+
sample_weight: OptionType(VectorType),
|
|
86
|
+
/** Column indices that contain categorical features (0-indexed) */
|
|
87
|
+
categorical_features: OptionType(ArrayType(IntegerType)),
|
|
88
|
+
/** Max categories for one-hot encoding (default 4). Features with more categories use partition-based splits. */
|
|
89
|
+
max_cat_to_onehot: OptionType(IntegerType),
|
|
90
|
+
/** Max categories considered per split for partition-based method (default 64). */
|
|
91
|
+
max_cat_threshold: OptionType(IntegerType),
|
|
44
92
|
});
|
|
45
93
|
// ============================================================================
|
|
46
94
|
// Model Blob Types
|
|
@@ -57,6 +105,8 @@ export const XGBoostModelBlobType = VariantType({
|
|
|
57
105
|
data: BlobType,
|
|
58
106
|
/** Number of input features */
|
|
59
107
|
n_features: IntegerType,
|
|
108
|
+
/** Column indices of categorical features (for prediction) */
|
|
109
|
+
categorical_features: OptionType(ArrayType(IntegerType)),
|
|
60
110
|
}),
|
|
61
111
|
/** XGBoost classifier model */
|
|
62
112
|
xgboost_classifier: StructType({
|
|
@@ -66,8 +116,30 @@ export const XGBoostModelBlobType = VariantType({
|
|
|
66
116
|
n_features: IntegerType,
|
|
67
117
|
/** Number of classes */
|
|
68
118
|
n_classes: IntegerType,
|
|
119
|
+
/** Column indices of categorical features (for prediction) */
|
|
120
|
+
categorical_features: OptionType(ArrayType(IntegerType)),
|
|
121
|
+
}),
|
|
122
|
+
/** XGBoost quantile regressor (multiple models, one per quantile) */
|
|
123
|
+
xgboost_quantile: StructType({
|
|
124
|
+
/** Cloudpickle serialized dict of {quantile: model} */
|
|
125
|
+
data: BlobType,
|
|
126
|
+
/** Quantiles this model predicts */
|
|
127
|
+
quantiles: VectorType,
|
|
128
|
+
/** Number of input features */
|
|
129
|
+
n_features: IntegerType,
|
|
130
|
+
/** Column indices of categorical features (for prediction) */
|
|
131
|
+
categorical_features: OptionType(ArrayType(IntegerType)),
|
|
69
132
|
}),
|
|
70
133
|
});
|
|
134
|
+
/**
|
|
135
|
+
* Result from XGBoost quantile prediction.
|
|
136
|
+
*/
|
|
137
|
+
export const XGBoostQuantilePredictResultType = StructType({
|
|
138
|
+
/** Quantile values that were predicted */
|
|
139
|
+
quantiles: VectorType,
|
|
140
|
+
/** Predictions matrix: (n_samples x n_quantiles) */
|
|
141
|
+
predictions: MatrixType,
|
|
142
|
+
});
|
|
71
143
|
// ============================================================================
|
|
72
144
|
// Platform Functions
|
|
73
145
|
// ============================================================================
|
|
@@ -113,6 +185,26 @@ export const xgboost_predict_class = East.platform("xgboost_predict_class", [XGB
|
|
|
113
185
|
* @returns Probability matrix (n_samples x n_classes)
|
|
114
186
|
*/
|
|
115
187
|
export const xgboost_predict_proba = East.platform("xgboost_predict_proba", [XGBoostModelBlobType, MatrixType], MatrixType);
|
|
188
|
+
/**
|
|
189
|
+
* Train XGBoost quantile regression models.
|
|
190
|
+
*
|
|
191
|
+
* Trains one model per quantile using pinball loss (reg:quantileerror).
|
|
192
|
+
* This provides prediction intervals and uncertainty quantification.
|
|
193
|
+
*
|
|
194
|
+
* @param X - Feature matrix
|
|
195
|
+
* @param y - Target vector
|
|
196
|
+
* @param config - Quantile regression configuration (includes quantiles array)
|
|
197
|
+
* @returns Model blob containing trained quantile models
|
|
198
|
+
*/
|
|
199
|
+
export const xgboost_train_quantile = East.platform("xgboost_train_quantile", [MatrixType, VectorType, XGBoostQuantileConfigType], XGBoostModelBlobType);
|
|
200
|
+
/**
|
|
201
|
+
* Predict quantiles with trained XGBoost quantile regressor.
|
|
202
|
+
*
|
|
203
|
+
* @param model - Trained quantile model blob
|
|
204
|
+
* @param X - Feature matrix
|
|
205
|
+
* @returns Quantiles and predictions matrix (n_samples x n_quantiles)
|
|
206
|
+
*/
|
|
207
|
+
export const xgboost_predict_quantile = East.platform("xgboost_predict_quantile", [XGBoostModelBlobType, MatrixType], XGBoostQuantilePredictResultType);
|
|
116
208
|
// ============================================================================
|
|
117
209
|
// Grouped Export
|
|
118
210
|
// ============================================================================
|
|
@@ -128,6 +220,10 @@ export const XGBoostTypes = {
|
|
|
128
220
|
LabelVectorType,
|
|
129
221
|
/** XGBoost configuration type */
|
|
130
222
|
XGBoostConfigType,
|
|
223
|
+
/** XGBoost quantile configuration type */
|
|
224
|
+
XGBoostQuantileConfigType,
|
|
225
|
+
/** Quantile prediction result type */
|
|
226
|
+
XGBoostQuantilePredictResultType,
|
|
131
227
|
/** Model blob type for XGBoost models */
|
|
132
228
|
ModelBlobType: XGBoostModelBlobType,
|
|
133
229
|
};
|
|
@@ -155,6 +251,7 @@ export const XGBoostTypes = {
|
|
|
155
251
|
* reg_lambda: variant('none', null),
|
|
156
252
|
* random_state: variant('some', 42n),
|
|
157
253
|
* n_jobs: variant('none', null),
|
|
254
|
+
* sample_weight: variant('none', null),
|
|
158
255
|
* });
|
|
159
256
|
* return $.return(XGBoost.trainRegressor(X, y, config));
|
|
160
257
|
* });
|
|
@@ -165,12 +262,16 @@ export const XGBoost = {
|
|
|
165
262
|
trainRegressor: xgboost_train_regressor,
|
|
166
263
|
/** Train XGBoost classifier */
|
|
167
264
|
trainClassifier: xgboost_train_classifier,
|
|
265
|
+
/** Train XGBoost quantile regressor */
|
|
266
|
+
trainQuantile: xgboost_train_quantile,
|
|
168
267
|
/** Make predictions with regressor */
|
|
169
268
|
predict: xgboost_predict,
|
|
170
269
|
/** Predict class labels with classifier */
|
|
171
270
|
predictClass: xgboost_predict_class,
|
|
172
271
|
/** Get class probabilities from classifier */
|
|
173
272
|
predictProba: xgboost_predict_proba,
|
|
273
|
+
/** Predict quantiles with quantile regressor */
|
|
274
|
+
predictQuantile: xgboost_predict_quantile,
|
|
174
275
|
/** Type definitions */
|
|
175
276
|
Types: XGBoostTypes,
|
|
176
277
|
};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"xgboost.js","sourceRoot":"","sources":["../../src/xgboost/xgboost.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EACH,IAAI,EACJ,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,QAAQ,GACX,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAEtE,yCAAyC;AACzC,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAEtE,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,UAAU,CAAC;IACxC,8CAA8C;IAC9C,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,qCAAqC;IACrC,SAAS,EAAE,UAAU,CAAC,WAAW,CAAC;IAClC,wDAAwD;IACxD,aAAa,EAAE,UAAU,CAAC,SAAS,CAAC;IACpC,mEAAmE;IACnE,gBAAgB,EAAE,UAAU,CAAC,WAAW,CAAC;IACzC,0DAA0D;IAC1D,SAAS,EAAE,UAAU,CAAC,SAAS,CAAC;IAChC,uEAAuE;IACvE,gBAAgB,EAAE,UAAU,CAAC,SAAS,CAAC;IACvC,yCAAyC;IACzC,SAAS,EAAE,UAAU,CAAC,SAAS,CAAC;IAChC,yCAAyC;IACzC,UAAU,EAAE,UAAU,CAAC,SAAS,CAAC;IACjC,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,4DAA4D;IAC5D,MAAM,EAAE,UAAU,CAAC,WAAW,CAAC;
|
|
1
|
+
{"version":3,"file":"xgboost.js","sourceRoot":"","sources":["../../src/xgboost/xgboost.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EACH,IAAI,EACJ,UAAU,EACV,WAAW,EACX,UAAU,EACV,SAAS,EACT,WAAW,EACX,SAAS,EACT,QAAQ,GACX,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAEtE,yCAAyC;AACzC,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAEtE,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,UAAU,CAAC;IACxC,8CAA8C;IAC9C,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,qCAAqC;IACrC,SAAS,EAAE,UAAU,CAAC,WAAW,CAAC;IAClC,wDAAwD;IACxD,aAAa,EAAE,UAAU,CAAC,SAAS,CAAC;IACpC,mEAAmE;IACnE,gBAAgB,EAAE,UAAU,CAAC,WAAW,CAAC;IACzC,0DAA0D;IAC1D,SAAS,EAAE,UAAU,CAAC,SAAS,CAAC;IAChC,uEAAuE;IACvE,gBAAgB,EAAE,UAAU,CAAC,SAAS,CAAC;IACvC,yCAAyC;IACzC,SAAS,EAAE,UAAU,CAAC,SAAS,CAAC;IAChC,yCAAyC;IACzC,UAAU,EAAE,UAAU,CAAC,SAAS,CAAC;IACjC,8EAA8E;IAC9E,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC;IAC5B,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,4DAA4D;IAC5D,MAAM,EAAE,UAAU,CAAC,WAAW,CAAC;IAC/B,oEAAoE;IACpE,aAAa,EAAE,UAAU,CAAC,UAAU,CAAC;IACrC,mEAAmE;IACnE,oBAAoB,EAAE,UAAU,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IACxD,iHAAiH;IACjH,iBAAiB,EAAE,UAAU,CAAC,WAAW,CAAC;IAC1C,mFAAmF;IACnF,iBAAiB,EAAE,UAAU,CAAC,WAAW,CAAC;CAC7C,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,UAAU,CAAC;IAChD,6EAA6E;IAC7E,SAAS,EAAE,UAAU;IACrB,8CAA8C;IAC9C,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,qCAAqC;IACrC,SAAS,EAAE,UAAU,CAAC,WAAW,CAAC;IAClC,wDAAwD;IACxD,aAAa,EAAE,UAAU,CAAC,SAAS,CAAC;IACpC,mEAAmE;IACnE,gBAAgB,EAAE,UAAU,CAAC,WAAW,CAAC;IACzC,0DAA0D;IAC1D,SAAS,EAAE,UAAU,CAAC,SAAS,CAAC;IAChC,uEAAuE;IACvE,gBAAgB,EAAE,UAAU,CAAC,SAAS,CAAC;IACvC,yCAAyC;IACzC,SAAS,EAAE,UAAU,CAAC,SAAS,CAAC;IAChC,yCAAyC;IACzC,UAAU,EAAE,UAAU,CAAC,SAAS,CAAC;IACjC,8EAA8E;IAC9E,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC;IAC5B,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,4DAA4D;IAC5D,MAAM,EAAE,UAAU,CAAC,WAAW,CAAC;IAC/B,oEAAoE;IACpE,aAAa,EAAE,UAAU,CAAC,UAAU,CAAC;IACrC,mEAAmE;IACnE,oBAAoB,EAAE,UAAU,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IACxD,iHAAiH;IACjH,iBAAiB,EAAE,UAAU,CAAC,WAAW,CAAC;IAC1C,mFAAmF;IACnF,iBAAiB,EAAE,UAAU,CAAC,WAAW,CAAC;CAC7C,CAAC,CAAC;AAEH,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;GAIG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,WAAW,CAAC;IAC5C,8BAA8B;IAC9B,iBAAiB,EAAE,UAAU,CAAC;QAC1B,mCAAmC;QACnC,IAAI,EAAE,QAAQ;QACd,+BAA+B;QAC/B,UAAU,EAAE,WAAW;QACvB,8DAA8D;QAC9D,oBAAoB,EAAE,UAAU,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;KAC3D,CAAC;IACF,+BAA+B;IAC/B,kBAAkB,EAAE,UAAU,CAAC;QAC3B,mCAAmC;QACnC,IAAI,EAAE,QAAQ;QACd,+BAA+B;QAC/B,UAAU,EAAE,WAAW;QACvB,wBAAwB;QACxB,SAAS,EAAE,WAAW;QACtB,8DAA8D;QAC9D,oBAAoB,EAAE,UAAU,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;KAC3D,CAAC;IACF,qEAAqE;IACrE,gBAAgB,EAAE,UAAU,CAAC;QACzB,uDAAuD;QACvD,IAAI,EAAE,QAAQ;QACd,oCAAoC;QACpC,SAAS,EAAE,UAAU;QACrB,+BAA+B;QAC/B,UAAU,EAAE,WAAW;QACvB,8DAA8D;QAC9D,oBAAoB,EAAE,UAAU,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;KAC3D,CAAC;CACL,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,gCAAgC,GAAG,UAAU,CAAC;IACvD,0CAA0C;IAC1C,SAAS,EAAE,UAAU;IACrB,oDAAoD;IACpD,WAAW,EAAE,UAAU;CAC1B,CAAC,CAAC;AAEH,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAE/E;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,IAAI,CAAC,QAAQ,CAChD,yBAAyB,EACzB,CAAC,UAAU,EAAE,UAAU,EAAE,iBAAiB,CAAC,EAC3C,oBAAoB,CACvB,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,wBAAwB,GAAG,IAAI,CAAC,QAAQ,CACjD,0BAA0B,EAC1B,CAAC,UAAU,EAAE,eAAe,EAAE,iBAAiB,CAAC,EAChD,oBAAoB,CACvB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CACxC,iBAAiB,EACjB,CAAC,oBAAoB,EAAE,UAAU,CAAC,EAClC,UAAU,CACb,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,IAAI,CAAC,QAAQ,CAC9C,uBAAuB,EACvB,CAAC,oBAAoB,EAAE,UAAU,CAAC,EAClC,eAAe,CAClB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,IAAI,CAAC,QAAQ,CAC9C,uBAAuB,EACvB,CAAC,oBAAoB,EAAE,UAAU,CAAC,EAClC,UAAU,CACb,CAAC;AAEF;;;;;;;;;;GAUG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG,IAAI,CAAC,QAAQ,CAC/C,wBAAwB,EACxB,CAAC,UAAU,EAAE,UAAU,EAAE,yBAAyB,CAAC,EACnD,oBAAoB,CACvB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,wBAAwB,GAAG,IAAI,CAAC,QAAQ,CACjD,0BAA0B,EAC1B,CAAC,oBAAoB,EAAE,UAAU,CAAC,EAClC,gCAAgC,CACnC,CAAC;AAEF,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG;IACxB,oCAAoC;IACpC,UAAU;IACV,uCAAuC;IACvC,UAAU;IACV,4CAA4C;IAC5C,eAAe;IACf,iCAAiC;IACjC,iBAAiB;IACjB,0CAA0C;IAC1C,yBAAyB;IACzB,sCAAsC;IACtC,gCAAgC;IAChC,yCAAyC;IACzC,aAAa,EAAE,oBAAoB;CAC7B,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,MAAM,CAAC,MAAM,OAAO,GAAG;IACnB,8BAA8B;IAC9B,cAAc,EAAE,uBAAuB;IACvC,+BAA+B;IAC/B,eAAe,EAAE,wBAAwB;IACzC,uCAAuC;IACvC,aAAa,EAAE,sBAAsB;IACrC,sCAAsC;IACtC,OAAO,EAAE,eAAe;IACxB,2CAA2C;IAC3C,YAAY,EAAE,qBAAqB;IACnC,8CAA8C;IAC9C,YAAY,EAAE,qBAAqB;IACnC,gDAAgD;IAChD,eAAe,EAAE,wBAAwB;IACzC,uBAAuB;IACvB,KAAK,EAAE,YAAY;CACb,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@elaraai/east-py-datascience",
|
|
3
|
-
"version": "0.0.2-beta.
|
|
3
|
+
"version": "0.0.2-beta.61",
|
|
4
4
|
"description": "East Data Science - ML/optimization platform functions for East (TypeScript definitions)",
|
|
5
5
|
"main": "dist/index.js",
|
|
6
6
|
"types": "dist/index.d.ts",
|
|
@@ -75,8 +75,8 @@
|
|
|
75
75
|
"eslint-plugin-headers": "^1.3.3",
|
|
76
76
|
"typescript": "~5.9.2"
|
|
77
77
|
},
|
|
78
|
-
"
|
|
79
|
-
"@elaraai/east": "^0.0.1-beta.
|
|
80
|
-
"@elaraai/east-node-std": "^0.0.1-beta.
|
|
78
|
+
"peerDependencies": {
|
|
79
|
+
"@elaraai/east": "^0.0.1-beta.31",
|
|
80
|
+
"@elaraai/east-node-std": "^0.0.1-beta.23"
|
|
81
81
|
}
|
|
82
82
|
}
|