@elaraai/east-py-datascience 0.0.2-beta.9 → 1.0.1

package/dist/xgboost/xgboost.js

@@ -1,177 +0,0 @@
-/**
- * Copyright (c) 2025 Elara AI Pty Ltd
- * Dual-licensed under AGPL-3.0 and commercial license. See LICENSE for details.
- */
-/**
- * XGBoost platform functions for East.
- *
- * Provides gradient boosting for regression and classification.
- * Uses ONNX for model serialization to enable portable inference.
- *
- * @packageDocumentation
- */
-import { East, StructType, VariantType, OptionType, IntegerType, FloatType, BlobType, } from "@elaraai/east";
-import { VectorType, MatrixType, LabelVectorType } from "../types.js";
-// Re-export shared types for convenience
-export { VectorType, MatrixType, LabelVectorType } from "../types.js";
-// ============================================================================
-// Config Types
-// ============================================================================
-/**
- * Configuration for XGBoost models.
- */
-export const XGBoostConfigType = StructType({
-    /** 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),
-});
-// ============================================================================
-// Model Blob Types
-// ============================================================================
-/**
- * Model blob type for serialized XGBoost models.
- *
- * Each model type has its own variant case containing cloudpickle bytes and metadata.
- */
-export const XGBoostModelBlobType = VariantType({
-    /** XGBoost regressor model */
-    xgboost_regressor: StructType({
-        /** Cloudpickle serialized model */
-        data: BlobType,
-        /** Number of input features */
-        n_features: IntegerType,
-    }),
-    /** XGBoost classifier model */
-    xgboost_classifier: StructType({
-        /** Cloudpickle serialized model */
-        data: BlobType,
-        /** Number of input features */
-        n_features: IntegerType,
-        /** Number of classes */
-        n_classes: IntegerType,
-    }),
-});
-// ============================================================================
-// Platform Functions
-// ============================================================================
-/**
- * Train an XGBoost regression model.
- *
- * @param X - Feature matrix
- * @param y - Target vector
- * @param config - XGBoost configuration
- * @returns Model blob containing trained regressor
- */
-export const xgboost_train_regressor = East.platform("xgboost_train_regressor", [MatrixType, VectorType, XGBoostConfigType], XGBoostModelBlobType);
-/**
- * Train an XGBoost classification model.
- *
- * @param X - Feature matrix
- * @param y - Label vector (integer class labels)
- * @param config - XGBoost configuration
- * @returns Model blob containing trained classifier
- */
-export const xgboost_train_classifier = East.platform("xgboost_train_classifier", [MatrixType, LabelVectorType, XGBoostConfigType], XGBoostModelBlobType);
-/**
- * Make predictions with a trained XGBoost regressor.
- *
- * @param model - Trained regressor model blob
- * @param X - Feature matrix
- * @returns Predicted values
- */
-export const xgboost_predict = East.platform("xgboost_predict", [XGBoostModelBlobType, MatrixType], VectorType);
-/**
- * Predict class labels with a trained XGBoost classifier.
- *
- * @param model - Trained classifier model blob
- * @param X - Feature matrix
- * @returns Predicted class labels
- */
-export const xgboost_predict_class = East.platform("xgboost_predict_class", [XGBoostModelBlobType, MatrixType], LabelVectorType);
-/**
- * Get class probabilities from a trained XGBoost classifier.
- *
- * @param model - Trained classifier model blob
- * @param X - Feature matrix
- * @returns Probability matrix (n_samples x n_classes)
- */
-export const xgboost_predict_proba = East.platform("xgboost_predict_proba", [XGBoostModelBlobType, MatrixType], MatrixType);
-// ============================================================================
-// Grouped Export
-// ============================================================================
-/**
- * Type definitions for XGBoost functions.
- */
-export const XGBoostTypes = {
-    /** Vector type (array of floats) */
-    VectorType,
-    /** Matrix type (2D array of floats) */
-    MatrixType,
-    /** Label vector type (array of integers) */
-    LabelVectorType,
-    /** XGBoost configuration type */
-    XGBoostConfigType,
-    /** Model blob type for XGBoost models */
-    ModelBlobType: XGBoostModelBlobType,
-};
-/**
- * XGBoost gradient boosting.
- *
- * Provides regression and classification with gradient boosted decision trees.
- *
- * @example
- * ```ts
- * import { East, variant } from "@elaraai/east";
- * import { XGBoost } from "@elaraai/east-py-datascience";
- *
- * const train = East.function([], XGBoost.Types.ModelBlobType, $ => {
- *     const X = $.let([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0], [7.0, 8.0]]);
- *     const y = $.let([1.0, 2.0, 3.0, 4.0]);
- *     const config = $.let({
- *         n_estimators: variant('some', 100n),
- *         max_depth: variant('some', 3n),
- *         learning_rate: variant('some', 0.1),
- *         min_child_weight: variant('none', null),
- *         subsample: variant('none', null),
- *         colsample_bytree: variant('none', null),
- *         reg_alpha: variant('none', null),
- *         reg_lambda: variant('none', null),
- *         random_state: variant('some', 42n),
- *         n_jobs: variant('none', null),
- *     });
- *     return $.return(XGBoost.trainRegressor(X, y, config));
- * });
- * ```
- */
-export const XGBoost = {
-    /** Train XGBoost regressor */
-    trainRegressor: xgboost_train_regressor,
-    /** Train XGBoost classifier */
-    trainClassifier: xgboost_train_classifier,
-    /** 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,
-    /** Type definitions */
-    Types: XGBoostTypes,
-};
-//# sourceMappingURL=xgboost.js.map

package/dist/xgboost/xgboost.js.map

@@ -1 +0,0 @@
-{"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"}