@elaraai/east-py-datascience 0.0.2-beta.1 → 0.0.2-beta.11

package/dist/xgboost/xgboost.js

@@ -41,6 +41,38 @@ export const XGBoostConfigType = StructType({
     random_state: OptionType(IntegerType),
     /** Number of parallel threads (default -1 for all cores) */
     n_jobs: OptionType(IntegerType),
+    /** Sample weights for training (one per sample, default uniform) */
+    sample_weight: OptionType(VectorType),
+});
+/**
+ * Configuration for XGBoost quantile regression.
+ * Trains separate models for each quantile.
+ */
+export const XGBoostQuantileConfigType = StructType({
+    /** Quantiles to predict (e.g., [0.1, 0.5, 0.9] for 80% interval + median) */
+    quantiles: VectorType,
+    /** Number of boosting rounds (default 100) */
+    n_estimators: OptionType(IntegerType),
+    /** Maximum tree depth (default 6) */
+    max_depth: OptionType(IntegerType),
+    /** Learning rate / step size shrinkage (default 0.3) */
+    learning_rate: OptionType(FloatType),
+    /** Minimum sum of instance weight needed in a child (default 1) */
+    min_child_weight: OptionType(IntegerType),
+    /** Subsample ratio of training instances (default 1.0) */
+    subsample: OptionType(FloatType),
+    /** Subsample ratio of columns when constructing trees (default 1.0) */
+    colsample_bytree: OptionType(FloatType),
+    /** L1 regularization term (default 0) */
+    reg_alpha: OptionType(FloatType),
+    /** L2 regularization term (default 1) */
+    reg_lambda: OptionType(FloatType),
+    /** Random seed for reproducibility */
+    random_state: OptionType(IntegerType),
+    /** Number of parallel threads (default -1 for all cores) */
+    n_jobs: OptionType(IntegerType),
+    /** Sample weights for training (one per sample, default uniform) */
+    sample_weight: OptionType(VectorType),
 });
 // ============================================================================
 // Model Blob Types
@@ -67,6 +99,24 @@ export const XGBoostModelBlobType = VariantType({
         /** Number of classes */
         n_classes: IntegerType,
     }),
+    /** XGBoost quantile regressor (multiple models, one per quantile) */
+    xgboost_quantile: StructType({
+        /** Cloudpickle serialized dict of {quantile: model} */
+        data: BlobType,
+        /** Quantiles this model predicts */
+        quantiles: VectorType,
+        /** Number of input features */
+        n_features: IntegerType,
+    }),
+});
+/**
+ * Result from XGBoost quantile prediction.
+ */
+export const XGBoostQuantilePredictResultType = StructType({
+    /** Quantile values that were predicted */
+    quantiles: VectorType,
+    /** Predictions matrix: (n_samples x n_quantiles) */
+    predictions: MatrixType,
 });
 // ============================================================================
 // Platform Functions
@@ -113,6 +163,26 @@ export const xgboost_predict_class = East.platform("xgboost_predict_class", [XGB
  * @returns Probability matrix (n_samples x n_classes)
  */
 export const xgboost_predict_proba = East.platform("xgboost_predict_proba", [XGBoostModelBlobType, MatrixType], MatrixType);
+/**
+ * Train XGBoost quantile regression models.
+ *
+ * Trains one model per quantile using pinball loss (reg:quantileerror).
+ * This provides prediction intervals and uncertainty quantification.
+ *
+ * @param X - Feature matrix
+ * @param y - Target vector
+ * @param config - Quantile regression configuration (includes quantiles array)
+ * @returns Model blob containing trained quantile models
+ */
+export const xgboost_train_quantile = East.platform("xgboost_train_quantile", [MatrixType, VectorType, XGBoostQuantileConfigType], XGBoostModelBlobType);
+/**
+ * Predict quantiles with trained XGBoost quantile regressor.
+ *
+ * @param model - Trained quantile model blob
+ * @param X - Feature matrix
+ * @returns Quantiles and predictions matrix (n_samples x n_quantiles)
+ */
+export const xgboost_predict_quantile = East.platform("xgboost_predict_quantile", [XGBoostModelBlobType, MatrixType], XGBoostQuantilePredictResultType);
 // ============================================================================
 // Grouped Export
 // ============================================================================
@@ -128,6 +198,10 @@ export const XGBoostTypes = {
     LabelVectorType,
     /** XGBoost configuration type */
     XGBoostConfigType,
+    /** XGBoost quantile configuration type */
+    XGBoostQuantileConfigType,
+    /** Quantile prediction result type */
+    XGBoostQuantilePredictResultType,
     /** Model blob type for XGBoost models */
     ModelBlobType: XGBoostModelBlobType,
 };
@@ -155,6 +229,7 @@ export const XGBoostTypes = {
  *         reg_lambda: variant('none', null),
  *         random_state: variant('some', 42n),
  *         n_jobs: variant('none', null),
+ *         sample_weight: variant('none', null),
  *     });
  *     return $.return(XGBoost.trainRegressor(X, y, config));
  * });
@@ -165,12 +240,16 @@ export const XGBoost = {
     trainRegressor: xgboost_train_regressor,
     /** Train XGBoost classifier */
     trainClassifier: xgboost_train_classifier,
+    /** Train XGBoost quantile regressor */
+    trainQuantile: xgboost_train_quantile,
     /** Make predictions with regressor */
     predict: xgboost_predict,
     /** Predict class labels with classifier */
     predictClass: xgboost_predict_class,
     /** Get class probabilities from classifier */
     predictProba: xgboost_predict_proba,
+    /** Predict quantiles with quantile regressor */
+    predictQuantile: xgboost_predict_quantile,
     /** Type definitions */
     Types: XGBoostTypes,
 };

package/dist/xgboost/xgboost.js.map

@@ -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;CAClC,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;KAC1B,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;KACzB,CAAC;CACL,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,+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,yCAAyC;IACzC,aAAa,EAAE,oBAAoB;CAC7B,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4BG;AACH,MAAM,CAAC,MAAM,OAAO,GAAG;IACnB,8BAA8B;IAC9B,cAAc,EAAE,uBAAuB;IACvC,+BAA+B;IAC/B,eAAe,EAAE,wBAAwB;IACzC,sCAAsC;IACtC,OAAO,EAAE,eAAe;IACxB,2CAA2C;IAC3C,YAAY,EAAE,qBAAqB;IACnC,8CAA8C;IAC9C,YAAY,EAAE,qBAAqB;IACnC,uBAAuB;IACvB,KAAK,EAAE,YAAY;CACb,CAAC"}
+{"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;IAC/B,oEAAoE;IACpE,aAAa,EAAE,UAAU,CAAC,UAAU,CAAC;CACxC,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,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;CACxC,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;KAC1B,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;KACzB,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;KAC1B,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

@@ -1,6 +1,6 @@
 {
   "name": "@elaraai/east-py-datascience",
-  "version": "0.0.2-beta.1",
+  "version": "0.0.2-beta.11",
   "description": "East Data Science - ML/optimization platform functions for East (TypeScript definitions)",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -76,7 +76,7 @@
     "typescript": "~5.9.2"
   },
   "dependencies": {
-    "@elaraai/east": "^0.0.1-beta.5",
-    "@elaraai/east-node-std": "^0.0.1-beta.4"
+    "@elaraai/east": "^0.0.1-beta.16",
+    "@elaraai/east-node-std": "^0.0.1-beta.9"
   }
 }