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

package/dist/ngboost/ngboost.d.ts

@@ -0,0 +1,433 @@
+/**
+ * Copyright (c) 2025 Elara AI Pty Ltd
+ * Dual-licensed under AGPL-3.0 and commercial license. See LICENSE for details.
+ */
+/**
+ * NGBoost platform functions for East.
+ *
+ * Provides probabilistic predictions with natural gradient boosting.
+ * Returns mean, standard deviation, and confidence intervals.
+ * Uses cloudpickle for model serialization.
+ *
+ * @packageDocumentation
+ */
+import { StructType, VariantType, OptionType, IntegerType, FloatType, BlobType } from "@elaraai/east";
+export { VectorType, MatrixType } from "../types.js";
+/**
+ * Distribution type for NGBoost.
+ */
+export declare const NGBoostDistributionType: VariantType<{
+    /** Normal (Gaussian) distribution */
+    normal: StructType<{}>;
+    /** Log-normal distribution (for positive targets) */
+    lognormal: StructType<{}>;
+}>;
+/**
+ * Configuration for NGBoost models.
+ */
+export declare const NGBoostConfigType: StructType<{
+    /** Number of boosting rounds (default 500) */
+    n_estimators: OptionType<IntegerType>;
+    /** Learning rate / step size shrinkage (default 0.01) */
+    learning_rate: OptionType<FloatType>;
+    /** Fraction of samples to use in each iteration (default 1.0) */
+    minibatch_frac: OptionType<FloatType>;
+    /** Fraction of features to use in each iteration (default 1.0) */
+    col_sample: OptionType<FloatType>;
+    /** Random seed for reproducibility */
+    random_state: OptionType<IntegerType>;
+    /** Distribution type (default normal) */
+    distribution: OptionType<VariantType<{
+        /** Normal (Gaussian) distribution */
+        normal: StructType<{}>;
+        /** Log-normal distribution (for positive targets) */
+        lognormal: StructType<{}>;
+    }>>;
+}>;
+/**
+ * Configuration for NGBoost predictions with uncertainty.
+ */
+export declare const NGBoostPredictConfigType: StructType<{
+    /** Confidence level for intervals (default 0.95) */
+    confidence_level: OptionType<FloatType>;
+}>;
+/**
+ * Result type for probabilistic predictions.
+ */
+export declare const NGBoostPredictResultType: StructType<{
+    /** Point predictions (mean) */
+    predictions: import("@elaraai/east").ArrayType<FloatType>;
+    /** Standard deviation */
+    std: OptionType<import("@elaraai/east").ArrayType<FloatType>>;
+    /** Lower confidence interval */
+    lower: OptionType<import("@elaraai/east").ArrayType<FloatType>>;
+    /** Upper confidence interval */
+    upper: OptionType<import("@elaraai/east").ArrayType<FloatType>>;
+}>;
+/**
+ * Model blob type for serialized NGBoost models.
+ */
+export declare const NGBoostModelBlobType: VariantType<{
+    /** NGBoost regressor model */
+    ngboost_regressor: StructType<{
+        /** Cloudpickle serialized model */
+        data: BlobType;
+        /** Distribution type used */
+        distribution: VariantType<{
+            /** Normal (Gaussian) distribution */
+            normal: StructType<{}>;
+            /** Log-normal distribution (for positive targets) */
+            lognormal: StructType<{}>;
+        }>;
+        /** Number of input features */
+        n_features: IntegerType;
+    }>;
+}>;
+/**
+ * Train an NGBoost regression model with probabilistic output.
+ *
+ * @param X - Feature matrix
+ * @param y - Target vector
+ * @param config - NGBoost configuration
+ * @returns Model blob containing trained regressor
+ */
+export declare const ngboost_train_regressor: import("@elaraai/east").PlatformDefinition<[import("@elaraai/east").ArrayType<import("@elaraai/east").ArrayType<FloatType>>, import("@elaraai/east").ArrayType<FloatType>, StructType<{
+    /** Number of boosting rounds (default 500) */
+    n_estimators: OptionType<IntegerType>;
+    /** Learning rate / step size shrinkage (default 0.01) */
+    learning_rate: OptionType<FloatType>;
+    /** Fraction of samples to use in each iteration (default 1.0) */
+    minibatch_frac: OptionType<FloatType>;
+    /** Fraction of features to use in each iteration (default 1.0) */
+    col_sample: OptionType<FloatType>;
+    /** Random seed for reproducibility */
+    random_state: OptionType<IntegerType>;
+    /** Distribution type (default normal) */
+    distribution: OptionType<VariantType<{
+        /** Normal (Gaussian) distribution */
+        normal: StructType<{}>;
+        /** Log-normal distribution (for positive targets) */
+        lognormal: StructType<{}>;
+    }>>;
+}>], VariantType<{
+    /** NGBoost regressor model */
+    ngboost_regressor: StructType<{
+        /** Cloudpickle serialized model */
+        data: BlobType;
+        /** Distribution type used */
+        distribution: VariantType<{
+            /** Normal (Gaussian) distribution */
+            normal: StructType<{}>;
+            /** Log-normal distribution (for positive targets) */
+            lognormal: StructType<{}>;
+        }>;
+        /** Number of input features */
+        n_features: IntegerType;
+    }>;
+}>>;
+/**
+ * Make point predictions (mean) with a trained NGBoost regressor.
+ *
+ * @param model - Trained regressor model blob
+ * @param X - Feature matrix
+ * @returns Predicted values (mean of predictive distribution)
+ */
+export declare const ngboost_predict: import("@elaraai/east").PlatformDefinition<[VariantType<{
+    /** NGBoost regressor model */
+    ngboost_regressor: StructType<{
+        /** Cloudpickle serialized model */
+        data: BlobType;
+        /** Distribution type used */
+        distribution: VariantType<{
+            /** Normal (Gaussian) distribution */
+            normal: StructType<{}>;
+            /** Log-normal distribution (for positive targets) */
+            lognormal: StructType<{}>;
+        }>;
+        /** Number of input features */
+        n_features: IntegerType;
+    }>;
+}>, import("@elaraai/east").ArrayType<import("@elaraai/east").ArrayType<FloatType>>], import("@elaraai/east").ArrayType<FloatType>>;
+/**
+ * Get predictions with full uncertainty from NGBoost regressor.
+ *
+ * Returns mean, standard deviation, and confidence intervals.
+ *
+ * @param model - Trained regressor model blob
+ * @param X - Feature matrix
+ * @param config - Prediction configuration (confidence level)
+ * @returns Predictions with uncertainty estimates
+ */
+export declare const ngboost_predict_dist: import("@elaraai/east").PlatformDefinition<[VariantType<{
+    /** NGBoost regressor model */
+    ngboost_regressor: StructType<{
+        /** Cloudpickle serialized model */
+        data: BlobType;
+        /** Distribution type used */
+        distribution: VariantType<{
+            /** Normal (Gaussian) distribution */
+            normal: StructType<{}>;
+            /** Log-normal distribution (for positive targets) */
+            lognormal: StructType<{}>;
+        }>;
+        /** Number of input features */
+        n_features: IntegerType;
+    }>;
+}>, import("@elaraai/east").ArrayType<import("@elaraai/east").ArrayType<FloatType>>, StructType<{
+    /** Confidence level for intervals (default 0.95) */
+    confidence_level: OptionType<FloatType>;
+}>], StructType<{
+    /** Point predictions (mean) */
+    predictions: import("@elaraai/east").ArrayType<FloatType>;
+    /** Standard deviation */
+    std: OptionType<import("@elaraai/east").ArrayType<FloatType>>;
+    /** Lower confidence interval */
+    lower: OptionType<import("@elaraai/east").ArrayType<FloatType>>;
+    /** Upper confidence interval */
+    upper: OptionType<import("@elaraai/east").ArrayType<FloatType>>;
+}>>;
+/**
+ * Type definitions for NGBoost functions.
+ */
+export declare const NGBoostTypes: {
+    /** Vector type (array of floats) */
+    readonly VectorType: import("@elaraai/east").ArrayType<FloatType>;
+    /** Matrix type (2D array of floats) */
+    readonly MatrixType: import("@elaraai/east").ArrayType<import("@elaraai/east").ArrayType<FloatType>>;
+    /** Distribution type for NGBoost */
+    readonly NGBoostDistributionType: VariantType<{
+        /** Normal (Gaussian) distribution */
+        normal: StructType<{}>;
+        /** Log-normal distribution (for positive targets) */
+        lognormal: StructType<{}>;
+    }>;
+    /** NGBoost configuration type */
+    readonly NGBoostConfigType: StructType<{
+        /** Number of boosting rounds (default 500) */
+        n_estimators: OptionType<IntegerType>;
+        /** Learning rate / step size shrinkage (default 0.01) */
+        learning_rate: OptionType<FloatType>;
+        /** Fraction of samples to use in each iteration (default 1.0) */
+        minibatch_frac: OptionType<FloatType>;
+        /** Fraction of features to use in each iteration (default 1.0) */
+        col_sample: OptionType<FloatType>;
+        /** Random seed for reproducibility */
+        random_state: OptionType<IntegerType>;
+        /** Distribution type (default normal) */
+        distribution: OptionType<VariantType<{
+            /** Normal (Gaussian) distribution */
+            normal: StructType<{}>;
+            /** Log-normal distribution (for positive targets) */
+            lognormal: StructType<{}>;
+        }>>;
+    }>;
+    /** Prediction configuration type */
+    readonly NGBoostPredictConfigType: StructType<{
+        /** Confidence level for intervals (default 0.95) */
+        confidence_level: OptionType<FloatType>;
+    }>;
+    /** Prediction result type */
+    readonly NGBoostPredictResultType: StructType<{
+        /** Point predictions (mean) */
+        predictions: import("@elaraai/east").ArrayType<FloatType>;
+        /** Standard deviation */
+        std: OptionType<import("@elaraai/east").ArrayType<FloatType>>;
+        /** Lower confidence interval */
+        lower: OptionType<import("@elaraai/east").ArrayType<FloatType>>;
+        /** Upper confidence interval */
+        upper: OptionType<import("@elaraai/east").ArrayType<FloatType>>;
+    }>;
+    /** Model blob type for NGBoost models */
+    readonly ModelBlobType: VariantType<{
+        /** NGBoost regressor model */
+        ngboost_regressor: StructType<{
+            /** Cloudpickle serialized model */
+            data: BlobType;
+            /** Distribution type used */
+            distribution: VariantType<{
+                /** Normal (Gaussian) distribution */
+                normal: StructType<{}>;
+                /** Log-normal distribution (for positive targets) */
+                lognormal: StructType<{}>;
+            }>;
+            /** Number of input features */
+            n_features: IntegerType;
+        }>;
+    }>;
+};
+/**
+ * NGBoost probabilistic gradient boosting.
+ *
+ * Provides regression with uncertainty quantification using natural gradient boosting.
+ *
+ * @example
+ * ```ts
+ * import { East, variant } from "@elaraai/east";
+ * import { NGBoost } from "@elaraai/east-py-datascience";
+ *
+ * const train = East.function([], NGBoost.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),
+ *         learning_rate: variant('some', 0.01),
+ *         minibatch_frac: variant('none', null),
+ *         col_sample: variant('none', null),
+ *         random_state: variant('some', 42n),
+ *         distribution: variant('none', null),
+ *     });
+ *     return $.return(NGBoost.trainRegressor(X, y, config));
+ * });
+ * ```
+ */
+export declare const NGBoost: {
+    /** Train NGBoost regressor */
+    readonly trainRegressor: import("@elaraai/east").PlatformDefinition<[import("@elaraai/east").ArrayType<import("@elaraai/east").ArrayType<FloatType>>, import("@elaraai/east").ArrayType<FloatType>, StructType<{
+        /** Number of boosting rounds (default 500) */
+        n_estimators: OptionType<IntegerType>;
+        /** Learning rate / step size shrinkage (default 0.01) */
+        learning_rate: OptionType<FloatType>;
+        /** Fraction of samples to use in each iteration (default 1.0) */
+        minibatch_frac: OptionType<FloatType>;
+        /** Fraction of features to use in each iteration (default 1.0) */
+        col_sample: OptionType<FloatType>;
+        /** Random seed for reproducibility */
+        random_state: OptionType<IntegerType>;
+        /** Distribution type (default normal) */
+        distribution: OptionType<VariantType<{
+            /** Normal (Gaussian) distribution */
+            normal: StructType<{}>;
+            /** Log-normal distribution (for positive targets) */
+            lognormal: StructType<{}>;
+        }>>;
+    }>], VariantType<{
+        /** NGBoost regressor model */
+        ngboost_regressor: StructType<{
+            /** Cloudpickle serialized model */
+            data: BlobType;
+            /** Distribution type used */
+            distribution: VariantType<{
+                /** Normal (Gaussian) distribution */
+                normal: StructType<{}>;
+                /** Log-normal distribution (for positive targets) */
+                lognormal: StructType<{}>;
+            }>;
+            /** Number of input features */
+            n_features: IntegerType;
+        }>;
+    }>>;
+    /** Make point predictions with regressor */
+    readonly predict: import("@elaraai/east").PlatformDefinition<[VariantType<{
+        /** NGBoost regressor model */
+        ngboost_regressor: StructType<{
+            /** Cloudpickle serialized model */
+            data: BlobType;
+            /** Distribution type used */
+            distribution: VariantType<{
+                /** Normal (Gaussian) distribution */
+                normal: StructType<{}>;
+                /** Log-normal distribution (for positive targets) */
+                lognormal: StructType<{}>;
+            }>;
+            /** Number of input features */
+            n_features: IntegerType;
+        }>;
+    }>, import("@elaraai/east").ArrayType<import("@elaraai/east").ArrayType<FloatType>>], import("@elaraai/east").ArrayType<FloatType>>;
+    /** Get predictions with uncertainty */
+    readonly predictDist: import("@elaraai/east").PlatformDefinition<[VariantType<{
+        /** NGBoost regressor model */
+        ngboost_regressor: StructType<{
+            /** Cloudpickle serialized model */
+            data: BlobType;
+            /** Distribution type used */
+            distribution: VariantType<{
+                /** Normal (Gaussian) distribution */
+                normal: StructType<{}>;
+                /** Log-normal distribution (for positive targets) */
+                lognormal: StructType<{}>;
+            }>;
+            /** Number of input features */
+            n_features: IntegerType;
+        }>;
+    }>, import("@elaraai/east").ArrayType<import("@elaraai/east").ArrayType<FloatType>>, StructType<{
+        /** Confidence level for intervals (default 0.95) */
+        confidence_level: OptionType<FloatType>;
+    }>], StructType<{
+        /** Point predictions (mean) */
+        predictions: import("@elaraai/east").ArrayType<FloatType>;
+        /** Standard deviation */
+        std: OptionType<import("@elaraai/east").ArrayType<FloatType>>;
+        /** Lower confidence interval */
+        lower: OptionType<import("@elaraai/east").ArrayType<FloatType>>;
+        /** Upper confidence interval */
+        upper: OptionType<import("@elaraai/east").ArrayType<FloatType>>;
+    }>>;
+    /** Type definitions */
+    readonly Types: {
+        /** Vector type (array of floats) */
+        readonly VectorType: import("@elaraai/east").ArrayType<FloatType>;
+        /** Matrix type (2D array of floats) */
+        readonly MatrixType: import("@elaraai/east").ArrayType<import("@elaraai/east").ArrayType<FloatType>>;
+        /** Distribution type for NGBoost */
+        readonly NGBoostDistributionType: VariantType<{
+            /** Normal (Gaussian) distribution */
+            normal: StructType<{}>;
+            /** Log-normal distribution (for positive targets) */
+            lognormal: StructType<{}>;
+        }>;
+        /** NGBoost configuration type */
+        readonly NGBoostConfigType: StructType<{
+            /** Number of boosting rounds (default 500) */
+            n_estimators: OptionType<IntegerType>;
+            /** Learning rate / step size shrinkage (default 0.01) */
+            learning_rate: OptionType<FloatType>;
+            /** Fraction of samples to use in each iteration (default 1.0) */
+            minibatch_frac: OptionType<FloatType>;
+            /** Fraction of features to use in each iteration (default 1.0) */
+            col_sample: OptionType<FloatType>;
+            /** Random seed for reproducibility */
+            random_state: OptionType<IntegerType>;
+            /** Distribution type (default normal) */
+            distribution: OptionType<VariantType<{
+                /** Normal (Gaussian) distribution */
+                normal: StructType<{}>;
+                /** Log-normal distribution (for positive targets) */
+                lognormal: StructType<{}>;
+            }>>;
+        }>;
+        /** Prediction configuration type */
+        readonly NGBoostPredictConfigType: StructType<{
+            /** Confidence level for intervals (default 0.95) */
+            confidence_level: OptionType<FloatType>;
+        }>;
+        /** Prediction result type */
+        readonly NGBoostPredictResultType: StructType<{
+            /** Point predictions (mean) */
+            predictions: import("@elaraai/east").ArrayType<FloatType>;
+            /** Standard deviation */
+            std: OptionType<import("@elaraai/east").ArrayType<FloatType>>;
+            /** Lower confidence interval */
+            lower: OptionType<import("@elaraai/east").ArrayType<FloatType>>;
+            /** Upper confidence interval */
+            upper: OptionType<import("@elaraai/east").ArrayType<FloatType>>;
+        }>;
+        /** Model blob type for NGBoost models */
+        readonly ModelBlobType: VariantType<{
+            /** NGBoost regressor model */
+            ngboost_regressor: StructType<{
+                /** Cloudpickle serialized model */
+                data: BlobType;
+                /** Distribution type used */
+                distribution: VariantType<{
+                    /** Normal (Gaussian) distribution */
+                    normal: StructType<{}>;
+                    /** Log-normal distribution (for positive targets) */
+                    lognormal: StructType<{}>;
+                }>;
+                /** Number of input features */
+                n_features: IntegerType;
+            }>;
+        }>;
+    };
+};
+//# sourceMappingURL=ngboost.d.ts.map

package/dist/ngboost/ngboost.d.ts.map

@@ -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,EACX,MAAM,eAAe,CAAC;AAIvB,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;;;mIAkCtC,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;;IAEpC,uCAAuC;;IAEvC,oCAAoC;;QApIpC,qCAAqC;;QAErC,qDAAqD;;;IAoIrD,iCAAiC;;QAxHjC,8CAA8C;;QAE9C,yDAAyD;;QAEzD,iEAAiE;;QAEjE,kEAAkE;;QAElE,sCAAsC;;QAEtC,yCAAyC;;YAxBzC,qCAAqC;;YAErC,qDAAqD;;;;IAsIrD,oCAAoC;;QAxGpC,oDAAoD;;;IA0GpD,6BAA6B;;QA9F7B,+BAA+B;;QAE/B,yBAAyB;;QAEzB,gCAAgC;;QAEhC,gCAAgC;;;IA0FhC,yCAAyC;;QA9EzC,8BAA8B;;YAE1B,mCAAmC;;YAEnC,6BAA6B;;gBAlEjC,qCAAqC;;gBAErC,qDAAqD;;;YAkEjD,+BAA+B;;;;CA0E7B,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,eAAO,MAAM,OAAO;IAChB,8BAA8B;;QA5J9B,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;;;;IAwGnC,4CAA4C;;QA9G5C,8BAA8B;;YAE1B,mCAAmC;;YAEnC,6BAA6B;;gBAlEjC,qCAAqC;;gBAErC,qDAAqD;;;YAkEjD,+BAA+B;;;;IA0GnC,uCAAuC;;QAhHvC,8BAA8B;;YAE1B,mCAAmC;;YAEnC,6BAA6B;;gBAlEjC,qCAAqC;;gBAErC,qDAAqD;;;YAkEjD,+BAA+B;;;;QApCnC,oDAAoD;;;QAYpD,+BAA+B;;QAE/B,yBAAyB;;QAEzB,gCAAgC;;QAEhC,gCAAgC;;;IA8HhC,uBAAuB;;QAhDvB,oCAAoC;;QAEpC,uCAAuC;;QAEvC,oCAAoC;;YApIpC,qCAAqC;;YAErC,qDAAqD;;;QAoIrD,iCAAiC;;YAxHjC,8CAA8C;;YAE9C,yDAAyD;;YAEzD,iEAAiE;;YAEjE,kEAAkE;;YAElE,sCAAsC;;YAEtC,yCAAyC;;gBAxBzC,qCAAqC;;gBAErC,qDAAqD;;;;QAsIrD,oCAAoC;;YAxGpC,oDAAoD;;;QA0GpD,6BAA6B;;YA9F7B,+BAA+B;;YAE/B,yBAAyB;;YAEzB,gCAAgC;;YAEhC,gCAAgC;;;QA0FhC,yCAAyC;;YA9EzC,8BAA8B;;gBAE1B,mCAAmC;;gBAEnC,6BAA6B;;oBAlEjC,qCAAqC;;oBAErC,qDAAqD;;;gBAkEjD,+BAA+B;;;;;CA8G7B,CAAC"}

package/dist/ngboost/ngboost.js

@@ -0,0 +1,178 @@
+/**
+ * Copyright (c) 2025 Elara AI Pty Ltd
+ * Dual-licensed under AGPL-3.0 and commercial license. See LICENSE for details.
+ */
+/**
+ * NGBoost platform functions for East.
+ *
+ * Provides probabilistic predictions with natural gradient boosting.
+ * Returns mean, standard deviation, and confidence intervals.
+ * Uses cloudpickle for model serialization.
+ *
+ * @packageDocumentation
+ */
+import { East, StructType, VariantType, OptionType, IntegerType, FloatType, BlobType, } from "@elaraai/east";
+import { VectorType, MatrixType } from "../types.js";
+// Re-export shared types for convenience
+export { VectorType, MatrixType } from "../types.js";
+// ============================================================================
+// Enum Types
+// ============================================================================
+/**
+ * Distribution type for NGBoost.
+ */
+export const NGBoostDistributionType = VariantType({
+    /** Normal (Gaussian) distribution */
+    normal: StructType({}),
+    /** Log-normal distribution (for positive targets) */
+    lognormal: StructType({}),
+});
+// ============================================================================
+// Config Types
+// ============================================================================
+/**
+ * Configuration for NGBoost models.
+ */
+export const NGBoostConfigType = StructType({
+    /** Number of boosting rounds (default 500) */
+    n_estimators: OptionType(IntegerType),
+    /** Learning rate / step size shrinkage (default 0.01) */
+    learning_rate: OptionType(FloatType),
+    /** Fraction of samples to use in each iteration (default 1.0) */
+    minibatch_frac: OptionType(FloatType),
+    /** Fraction of features to use in each iteration (default 1.0) */
+    col_sample: OptionType(FloatType),
+    /** Random seed for reproducibility */
+    random_state: OptionType(IntegerType),
+    /** Distribution type (default normal) */
+    distribution: OptionType(NGBoostDistributionType),
+});
+/**
+ * Configuration for NGBoost predictions with uncertainty.
+ */
+export const NGBoostPredictConfigType = StructType({
+    /** Confidence level for intervals (default 0.95) */
+    confidence_level: OptionType(FloatType),
+});
+// ============================================================================
+// Result Types
+// ============================================================================
+/**
+ * Result type for probabilistic predictions.
+ */
+export const NGBoostPredictResultType = StructType({
+    /** Point predictions (mean) */
+    predictions: VectorType,
+    /** Standard deviation */
+    std: OptionType(VectorType),
+    /** Lower confidence interval */
+    lower: OptionType(VectorType),
+    /** Upper confidence interval */
+    upper: OptionType(VectorType),
+});
+// ============================================================================
+// Model Blob Types
+// ============================================================================
+/**
+ * Model blob type for serialized NGBoost models.
+ */
+export const NGBoostModelBlobType = VariantType({
+    /** NGBoost regressor model */
+    ngboost_regressor: StructType({
+        /** Cloudpickle serialized model */
+        data: BlobType,
+        /** Distribution type used */
+        distribution: NGBoostDistributionType,
+        /** Number of input features */
+        n_features: IntegerType,
+    }),
+});
+// ============================================================================
+// Platform Functions
+// ============================================================================
+/**
+ * Train an NGBoost regression model with probabilistic output.
+ *
+ * @param X - Feature matrix
+ * @param y - Target vector
+ * @param config - NGBoost configuration
+ * @returns Model blob containing trained regressor
+ */
+export const ngboost_train_regressor = East.platform("ngboost_train_regressor", [MatrixType, VectorType, NGBoostConfigType], NGBoostModelBlobType);
+/**
+ * Make point predictions (mean) with a trained NGBoost regressor.
+ *
+ * @param model - Trained regressor model blob
+ * @param X - Feature matrix
+ * @returns Predicted values (mean of predictive distribution)
+ */
+export const ngboost_predict = East.platform("ngboost_predict", [NGBoostModelBlobType, MatrixType], VectorType);
+/**
+ * Get predictions with full uncertainty from NGBoost regressor.
+ *
+ * Returns mean, standard deviation, and confidence intervals.
+ *
+ * @param model - Trained regressor model blob
+ * @param X - Feature matrix
+ * @param config - Prediction configuration (confidence level)
+ * @returns Predictions with uncertainty estimates
+ */
+export const ngboost_predict_dist = East.platform("ngboost_predict_dist", [NGBoostModelBlobType, MatrixType, NGBoostPredictConfigType], NGBoostPredictResultType);
+// ============================================================================
+// Grouped Export
+// ============================================================================
+/**
+ * Type definitions for NGBoost functions.
+ */
+export const NGBoostTypes = {
+    /** Vector type (array of floats) */
+    VectorType,
+    /** Matrix type (2D array of floats) */
+    MatrixType,
+    /** Distribution type for NGBoost */
+    NGBoostDistributionType,
+    /** NGBoost configuration type */
+    NGBoostConfigType,
+    /** Prediction configuration type */
+    NGBoostPredictConfigType,
+    /** Prediction result type */
+    NGBoostPredictResultType,
+    /** Model blob type for NGBoost models */
+    ModelBlobType: NGBoostModelBlobType,
+};
+/**
+ * NGBoost probabilistic gradient boosting.
+ *
+ * Provides regression with uncertainty quantification using natural gradient boosting.
+ *
+ * @example
+ * ```ts
+ * import { East, variant } from "@elaraai/east";
+ * import { NGBoost } from "@elaraai/east-py-datascience";
+ *
+ * const train = East.function([], NGBoost.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),
+ *         learning_rate: variant('some', 0.01),
+ *         minibatch_frac: variant('none', null),
+ *         col_sample: variant('none', null),
+ *         random_state: variant('some', 42n),
+ *         distribution: variant('none', null),
+ *     });
+ *     return $.return(NGBoost.trainRegressor(X, y, config));
+ * });
+ * ```
+ */
+export const NGBoost = {
+    /** Train NGBoost regressor */
+    trainRegressor: ngboost_train_regressor,
+    /** Make point predictions with regressor */
+    predict: ngboost_predict,
+    /** Get predictions with uncertainty */
+    predictDist: ngboost_predict_dist,
+    /** Type definitions */
+    Types: NGBoostTypes,
+};
+//# sourceMappingURL=ngboost.js.map

package/dist/ngboost/ngboost.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ngboost.js","sourceRoot":"","sources":["../../src/ngboost/ngboost.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;;GAQG;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,MAAM,aAAa,CAAC;AAErD,yCAAyC;AACzC,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,+EAA+E;AAC/E,aAAa;AACb,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,WAAW,CAAC;IAC/C,qCAAqC;IACrC,MAAM,EAAE,UAAU,CAAC,EAAE,CAAC;IACtB,qDAAqD;IACrD,SAAS,EAAE,UAAU,CAAC,EAAE,CAAC;CAC5B,CAAC,CAAC;AAEH,+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,yDAAyD;IACzD,aAAa,EAAE,UAAU,CAAC,SAAS,CAAC;IACpC,iEAAiE;IACjE,cAAc,EAAE,UAAU,CAAC,SAAS,CAAC;IACrC,kEAAkE;IAClE,UAAU,EAAE,UAAU,CAAC,SAAS,CAAC;IACjC,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,yCAAyC;IACzC,YAAY,EAAE,UAAU,CAAC,uBAAuB,CAAC;CACpD,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,wBAAwB,GAAG,UAAU,CAAC;IAC/C,oDAAoD;IACpD,gBAAgB,EAAE,UAAU,CAAC,SAAS,CAAC;CAC1C,CAAC,CAAC;AAEH,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,wBAAwB,GAAG,UAAU,CAAC;IAC/C,+BAA+B;IAC/B,WAAW,EAAE,UAAU;IACvB,yBAAyB;IACzB,GAAG,EAAE,UAAU,CAAC,UAAU,CAAC;IAC3B,gCAAgC;IAChC,KAAK,EAAE,UAAU,CAAC,UAAU,CAAC;IAC7B,gCAAgC;IAChC,KAAK,EAAE,UAAU,CAAC,UAAU,CAAC;CAChC,CAAC,CAAC;AAEH,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,WAAW,CAAC;IAC5C,8BAA8B;IAC9B,iBAAiB,EAAE,UAAU,CAAC;QAC1B,mCAAmC;QACnC,IAAI,EAAE,QAAQ;QACd,6BAA6B;QAC7B,YAAY,EAAE,uBAAuB;QACrC,+BAA+B;QAC/B,UAAU,EAAE,WAAW;KAC1B,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;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CACxC,iBAAiB,EACjB,CAAC,oBAAoB,EAAE,UAAU,CAAC,EAClC,UAAU,CACb,CAAC;AAEF;;;;;;;;;GASG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,IAAI,CAAC,QAAQ,CAC7C,sBAAsB,EACtB,CAAC,oBAAoB,EAAE,UAAU,EAAE,wBAAwB,CAAC,EAC5D,wBAAwB,CAC3B,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,oCAAoC;IACpC,uBAAuB;IACvB,iCAAiC;IACjC,iBAAiB;IACjB,oCAAoC;IACpC,wBAAwB;IACxB,6BAA6B;IAC7B,wBAAwB;IACxB,yCAAyC;IACzC,aAAa,EAAE,oBAAoB;CAC7B,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;;GAwBG;AACH,MAAM,CAAC,MAAM,OAAO,GAAG;IACnB,8BAA8B;IAC9B,cAAc,EAAE,uBAAuB;IACvC,4CAA4C;IAC5C,OAAO,EAAE,eAAe;IACxB,uCAAuC;IACvC,WAAW,EAAE,oBAAoB;IACjC,uBAAuB;IACvB,KAAK,EAAE,YAAY;CACb,CAAC"}