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

package/dist/xgboost/xgboost.js

@@ -0,0 +1,177 @@
+/**
+ * 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

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

package/package.json

@@ -0,0 +1,82 @@
+{
+  "name": "@elaraai/east-py-datascience",
+  "version": "0.0.2-beta.1",
+  "description": "East Data Science - ML/optimization platform functions for East (TypeScript definitions)",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    }
+  },
+  "files": [
+    "dist",
+    "!dist/**/*.spec.*",
+    "README.md",
+    "LICENSE.md",
+    "CONTRIBUTING.md",
+    "CLA.md"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "test": "npm run build && node --enable-source-maps --test 'dist/**/*.spec.js'",
+    "test:coverage": "npm run build && node --enable-source-maps --test --experimental-test-coverage 'dist/**/*.spec.js'",
+    "test:export": "npm run build && EXPORT_TEST_IR=/tmp/east-py-datascience npm test 2>&1 | grep 'Exported test IR'",
+    "lint": "eslint .",
+    "lint:fix": "eslint . --fix",
+    "version:patch": "npm version patch -m 'chore: bump version to %s'",
+    "version:minor": "npm version minor -m 'chore: bump version to %s'",
+    "version:major": "npm version major -m 'chore: bump version to %s'",
+    "version:prepatch": "npm version prepatch --preid=beta -m 'chore: bump version to %s'",
+    "version:preminor": "npm version preminor --preid=beta -m 'chore: bump version to %s'",
+    "version:premajor": "npm version premajor --preid=beta -m 'chore: bump version to %s'",
+    "version:prerelease": "npm version prerelease --preid=beta -m 'chore: bump version to %s'",
+    "version:patch:dry": "npm version patch --no-git-tag-version",
+    "version:minor:dry": "npm version minor --no-git-tag-version",
+    "version:major:dry": "npm version major --no-git-tag-version",
+    "version:prepatch:dry": "npm version prepatch --preid=beta --no-git-tag-version",
+    "version:preminor:dry": "npm version preminor --preid=beta --no-git-tag-version",
+    "version:premajor:dry": "npm version premajor --preid=beta --no-git-tag-version",
+    "version:prerelease:dry": "npm version prerelease --preid=beta --no-git-tag-version",
+    "release:patch": "npm run version:patch && git push && git push --tags",
+    "release:minor": "npm run version:minor && git push && git push --tags",
+    "release:major": "npm run version:major && git push && git push --tags",
+    "release:prepatch": "npm run version:prepatch && git push && git push --tags",
+    "release:preminor": "npm run version:preminor && git push && git push --tags",
+    "release:premajor": "npm run version:premajor && git push && git push --tags",
+    "release:prerelease": "npm run version:prerelease && git push && git push --tags"
+  },
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/elaraai/east-py.git",
+    "directory": "packages/east-py-datascience"
+  },
+  "keywords": [
+    "east",
+    "elara",
+    "datascience",
+    "ml",
+    "optimization",
+    "mads",
+    "pynomad"
+  ],
+  "author": "Elara AI Pty Ltd",
+  "license": "SEE LICENSE IN LICENSE",
+  "type": "module",
+  "engines": {
+    "node": ">=22.0.0"
+  },
+  "devDependencies": {
+    "@types/node": "^22.18.1",
+    "@typescript-eslint/eslint-plugin": "^8.42.0",
+    "@typescript-eslint/parser": "^8.42.0",
+    "eslint": "^9.34.0",
+    "eslint-plugin-headers": "^1.3.3",
+    "typescript": "~5.9.2"
+  },
+  "dependencies": {
+    "@elaraai/east": "^0.0.1-beta.5",
+    "@elaraai/east-node-std": "^0.0.1-beta.4"
+  }
+}