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

package/dist/gp/gp.js

@@ -0,0 +1,170 @@
+/**
+ * Copyright (c) 2025 Elara AI Pty Ltd
+ * Dual-licensed under AGPL-3.0 and commercial license. See LICENSE for details.
+ */
+/**
+ * Gaussian Process platform functions for East.
+ *
+ * Provides Gaussian Process regression using scikit-learn.
+ * Uses cloudpickle for model serialization.
+ *
+ * @packageDocumentation
+ */
+import { East, StructType, VariantType, OptionType, IntegerType, FloatType, BooleanType, BlobType, StringType, } from "@elaraai/east";
+import { VectorType, MatrixType } from "../types.js";
+// Re-export shared types for convenience
+export { VectorType, MatrixType } from "../types.js";
+// ============================================================================
+// Enum Types
+// ============================================================================
+/**
+ * Kernel type for Gaussian Process.
+ */
+export const GPKernelType = VariantType({
+    /** Radial Basis Function (squared exponential) */
+    rbf: StructType({}),
+    /** Matern with nu=1/2 (exponential) */
+    matern_1_2: StructType({}),
+    /** Matern with nu=3/2 */
+    matern_3_2: StructType({}),
+    /** Matern with nu=5/2 */
+    matern_5_2: StructType({}),
+    /** Rational Quadratic */
+    rational_quadratic: StructType({}),
+    /** Dot Product (linear) */
+    dot_product: StructType({}),
+});
+// ============================================================================
+// Config Types
+// ============================================================================
+/**
+ * Configuration for Gaussian Process Regressor.
+ */
+export const GPConfigType = StructType({
+    /** Kernel type (default rbf) */
+    kernel: OptionType(GPKernelType),
+    /** Noise level added to diagonal (default 1e-10) */
+    alpha: OptionType(FloatType),
+    /** Number of restarts for optimizer (default 0) */
+    n_restarts_optimizer: OptionType(IntegerType),
+    /** Whether to normalize target values (default false) */
+    normalize_y: OptionType(BooleanType),
+    /** Random seed for reproducibility */
+    random_state: OptionType(IntegerType),
+});
+// ============================================================================
+// Result Types
+// ============================================================================
+/**
+ * Result type for GP prediction with uncertainty.
+ */
+export const GPPredictResultType = StructType({
+    /** Predicted mean values */
+    mean: VectorType,
+    /** Predicted standard deviation (uncertainty) */
+    std: VectorType,
+});
+// ============================================================================
+// Model Blob Types
+// ============================================================================
+/**
+ * Model blob type for serialized GP models.
+ */
+export const GPModelBlobType = VariantType({
+    /** Gaussian Process Regressor */
+    gp_regressor: StructType({
+        /** Cloudpickle serialized model */
+        data: BlobType,
+        /** Number of input features */
+        n_features: IntegerType,
+        /** Kernel type name for reference */
+        kernel_type: StringType,
+    }),
+});
+// ============================================================================
+// Platform Functions
+// ============================================================================
+/**
+ * Train a Gaussian Process Regressor.
+ *
+ * @param X - Feature matrix
+ * @param y - Target vector
+ * @param config - GP configuration
+ * @returns Trained GP model blob
+ */
+export const gp_train = East.platform("gp_train", [MatrixType, VectorType, GPConfigType], GPModelBlobType);
+/**
+ * Make predictions with a trained Gaussian Process.
+ *
+ * Returns point predictions (mean only).
+ *
+ * @param model - Trained GP model blob
+ * @param X - Feature matrix
+ * @returns Predicted values
+ */
+export const gp_predict = East.platform("gp_predict", [GPModelBlobType, MatrixType], VectorType);
+/**
+ * Make predictions with uncertainty estimates.
+ *
+ * Returns both mean and standard deviation.
+ *
+ * @param model - Trained GP model blob
+ * @param X - Feature matrix
+ * @returns Prediction result with mean and std
+ */
+export const gp_predict_std = East.platform("gp_predict_std", [GPModelBlobType, MatrixType], GPPredictResultType);
+// ============================================================================
+// Grouped Export
+// ============================================================================
+/**
+ * Type definitions for GP functions.
+ */
+export const GPTypes = {
+    /** Vector type (array of floats) */
+    VectorType,
+    /** Matrix type (2D array of floats) */
+    MatrixType,
+    /** Kernel type */
+    GPKernelType,
+    /** Configuration type */
+    GPConfigType,
+    /** Prediction result type with uncertainty */
+    GPPredictResultType,
+    /** Model blob type for GP models */
+    ModelBlobType: GPModelBlobType,
+};
+/**
+ * Gaussian Process regression.
+ *
+ * Provides probabilistic regression with uncertainty quantification.
+ *
+ * @example
+ * ```ts
+ * import { East, variant } from "@elaraai/east";
+ * import { GP } from "@elaraai/east-py-datascience";
+ *
+ * const train = East.function([], GP.Types.ModelBlobType, $ => {
+ *     const X = $.let([[1.0], [2.0], [3.0], [4.0]]);
+ *     const y = $.let([1.0, 4.0, 9.0, 16.0]);
+ *     const config = $.let({
+ *         kernel: variant('some', variant('rbf', {})),
+ *         alpha: variant('some', 1e-10),
+ *         n_restarts_optimizer: variant('some', 5n),
+ *         normalize_y: variant('some', true),
+ *         random_state: variant('some', 42n),
+ *     });
+ *     return $.return(GP.train(X, y, config));
+ * });
+ * ```
+ */
+export const GP = {
+    /** Train GP regressor */
+    train: gp_train,
+    /** Make predictions (mean only) */
+    predict: gp_predict,
+    /** Make predictions with uncertainty */
+    predictStd: gp_predict_std,
+    /** Type definitions */
+    Types: GPTypes,
+};
+//# sourceMappingURL=gp.js.map

package/dist/gp/gp.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gp.js","sourceRoot":"","sources":["../../src/gp/gp.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EACH,IAAI,EACJ,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,WAAW,EACX,QAAQ,EACR,UAAU,GACb,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,YAAY,GAAG,WAAW,CAAC;IACpC,kDAAkD;IAClD,GAAG,EAAE,UAAU,CAAC,EAAE,CAAC;IACnB,uCAAuC;IACvC,UAAU,EAAE,UAAU,CAAC,EAAE,CAAC;IAC1B,yBAAyB;IACzB,UAAU,EAAE,UAAU,CAAC,EAAE,CAAC;IAC1B,yBAAyB;IACzB,UAAU,EAAE,UAAU,CAAC,EAAE,CAAC;IAC1B,yBAAyB;IACzB,kBAAkB,EAAE,UAAU,CAAC,EAAE,CAAC;IAClC,2BAA2B;IAC3B,WAAW,EAAE,UAAU,CAAC,EAAE,CAAC;CAC9B,CAAC,CAAC;AAEH,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG,UAAU,CAAC;IACnC,gCAAgC;IAChC,MAAM,EAAE,UAAU,CAAC,YAAY,CAAC;IAChC,oDAAoD;IACpD,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC;IAC5B,mDAAmD;IACnD,oBAAoB,EAAE,UAAU,CAAC,WAAW,CAAC;IAC7C,yDAAyD;IACzD,WAAW,EAAE,UAAU,CAAC,WAAW,CAAC;IACpC,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;CACxC,CAAC,CAAC;AAEH,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,UAAU,CAAC;IAC1C,4BAA4B;IAC5B,IAAI,EAAE,UAAU;IAChB,iDAAiD;IACjD,GAAG,EAAE,UAAU;CAClB,CAAC,CAAC;AAEH,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,WAAW,CAAC;IACvC,iCAAiC;IACjC,YAAY,EAAE,UAAU,CAAC;QACrB,mCAAmC;QACnC,IAAI,EAAE,QAAQ;QACd,+BAA+B;QAC/B,UAAU,EAAE,WAAW;QACvB,qCAAqC;QACrC,WAAW,EAAE,UAAU;KAC1B,CAAC;CACL,CAAC,CAAC;AAEH,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAE/E;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CACjC,UAAU,EACV,CAAC,UAAU,EAAE,UAAU,EAAE,YAAY,CAAC,EACtC,eAAe,CAClB,CAAC;AAEF;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CACnC,YAAY,EACZ,CAAC,eAAe,EAAE,UAAU,CAAC,EAC7B,UAAU,CACb,CAAC;AAEF;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,CACvC,gBAAgB,EAChB,CAAC,eAAe,EAAE,UAAU,CAAC,EAC7B,mBAAmB,CACtB,CAAC;AAEF,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,OAAO,GAAG;IACnB,oCAAoC;IACpC,UAAU;IACV,uCAAuC;IACvC,UAAU;IACV,kBAAkB;IAClB,YAAY;IACZ,yBAAyB;IACzB,YAAY;IACZ,8CAA8C;IAC9C,mBAAmB;IACnB,oCAAoC;IACpC,aAAa,EAAE,eAAe;CACxB,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAM,CAAC,MAAM,EAAE,GAAG;IACd,yBAAyB;IACzB,KAAK,EAAE,QAAQ;IACf,mCAAmC;IACnC,OAAO,EAAE,UAAU;IACnB,wCAAwC;IACxC,UAAU,EAAE,cAAc;IAC1B,uBAAuB;IACvB,KAAK,EAAE,OAAO;CACR,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,27 @@
+/**
+ * Copyright (c) 2025 Elara AI Pty Ltd
+ * Dual-licensed under AGPL-3.0 and commercial license. See LICENSE for details.
+ */
+/**
+ * East Data Science - ML and optimization platform functions for East.
+ *
+ * This package provides data science capabilities for East programs:
+ * - MADS: Derivative-free blackbox optimization (PyNomadBBO)
+ * - Optuna: Bayesian optimization with TPE sampler
+ * - SimAnneal: Discrete optimization with Simulated Annealing
+ *
+ * @packageDocumentation
+ */
+export { MADS, mads_optimize, MADSTypes, VectorType, ScalarObjectiveType, MADSBoundsType, MADSConstraintType, MADSDirectionType, MADSConfigType, MADSResultType, } from "./mads/mads.js";
+export { Optuna, optuna_optimize, OptunaTypes, ParamValueType, ParamSpaceKindType, ParamSpaceType, NamedParamType, OptimizationDirectionType, PrunerType, OptunaStudyConfigType, TrialResultType, StudyResultType, ObjectiveFunctionType, } from "./optuna/optuna.js";
+export { SimAnneal, simanneal_optimize, simanneal_optimize_permutation, simanneal_optimize_subset, SimAnnealTypes, DiscreteStateType, EnergyFunctionType, MoveFunctionType, PermutationEnergyType, SubsetEnergyType, AnnealConfigType, AnnealResultType, } from "./simanneal/simanneal.js";
+export { Sklearn, sklearn_train_test_split, sklearn_standard_scaler_fit, sklearn_standard_scaler_transform, sklearn_min_max_scaler_fit, sklearn_min_max_scaler_transform, sklearn_metrics_regression, sklearn_metrics_classification, sklearn_regressor_chain_train, sklearn_regressor_chain_predict, SklearnTypes, SplitConfigType, SplitResultType, RegressionMetricsType, ClassificationMetricsType, SklearnModelBlobType, MultiTargetType, RegressorChainBaseConfigType, RegressorChainConfigType, } from "./sklearn/sklearn.js";
+export { Scipy, scipy_curve_fit, scipy_stats_describe, scipy_stats_pearsonr, scipy_stats_spearmanr, scipy_interpolate_1d_fit, scipy_interpolate_1d_predict, scipy_optimize_minimize, scipy_optimize_minimize_quadratic, ScipyTypes, OptimizeMethodType, InterpolationKindType, OptimizeConfigType, InterpolateConfigType, ParamBoundsType, CustomCurveFunctionType, CurveFunctionType, CurveFitConfigType, QuadraticConfigType, StatsDescribeResultType, CorrelationResultType, CurveFitResultType, OptimizeResultType, ScipyModelBlobType, } from "./scipy/scipy.js";
+export { XGBoost, xgboost_train_regressor, xgboost_train_classifier, xgboost_predict, xgboost_predict_class, xgboost_predict_proba, XGBoostTypes, XGBoostConfigType, XGBoostModelBlobType, } from "./xgboost/xgboost.js";
+export { LightGBM, lightgbm_train_regressor, lightgbm_train_classifier, lightgbm_predict, lightgbm_predict_class, lightgbm_predict_proba, LightGBMTypes, LightGBMConfigType, LightGBMModelBlobType, } from "./lightgbm/lightgbm.js";
+export { NGBoost, ngboost_train_regressor, ngboost_predict, ngboost_predict_dist, NGBoostTypes, NGBoostDistributionType, NGBoostConfigType, NGBoostPredictConfigType, NGBoostPredictResultType, NGBoostModelBlobType, } from "./ngboost/ngboost.js";
+export { Shap, shap_tree_explainer_create, shap_kernel_explainer_create, shap_compute_values, shap_feature_importance, ShapTypes, ShapResultType, FeatureImportanceType, ShapModelBlobType, AnyModelBlobType, StringVectorType, } from "./shap/shap.js";
+export { Torch, torch_mlp_train, torch_mlp_predict, TorchTypes, TorchActivationType, TorchLossType, TorchOptimizerType, TorchMLPConfigType, TorchTrainConfigType, TorchTrainResultType, TorchTrainOutputType, TorchModelBlobType, } from "./torch/torch.js";
+export { GP, gp_train, gp_predict, gp_predict_std, GPTypes, GPKernelType, GPConfigType, GPPredictResultType, GPModelBlobType, } from "./gp/gp.js";
+export { VectorType as SharedVectorType, MatrixType as SharedMatrixType, ScalarObjectiveType as SharedScalarObjectiveType, VectorObjectiveType, LabelVectorType, } from "./types.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;;;GASG;AAGH,OAAO,EACH,IAAI,EACJ,aAAa,EACb,SAAS,EACT,UAAU,EACV,mBAAmB,EACnB,cAAc,EACd,kBAAkB,EAClB,iBAAiB,EACjB,cAAc,EACd,cAAc,GACjB,MAAM,gBAAgB,CAAC;AAGxB,OAAO,EACH,MAAM,EACN,eAAe,EACf,WAAW,EACX,cAAc,EACd,kBAAkB,EAClB,cAAc,EACd,cAAc,EACd,yBAAyB,EACzB,UAAU,EACV,qBAAqB,EACrB,eAAe,EACf,eAAe,EACf,qBAAqB,GACxB,MAAM,oBAAoB,CAAC;AAG5B,OAAO,EACH,SAAS,EACT,kBAAkB,EAClB,8BAA8B,EAC9B,yBAAyB,EACzB,cAAc,EACd,iBAAiB,EACjB,kBAAkB,EAClB,gBAAgB,EAChB,qBAAqB,EACrB,gBAAgB,EAChB,gBAAgB,EAChB,gBAAgB,GACnB,MAAM,0BAA0B,CAAC;AAGlC,OAAO,EACH,OAAO,EACP,wBAAwB,EACxB,2BAA2B,EAC3B,iCAAiC,EACjC,0BAA0B,EAC1B,gCAAgC,EAChC,0BAA0B,EAC1B,8BAA8B,EAC9B,6BAA6B,EAC7B,+BAA+B,EAC/B,YAAY,EACZ,eAAe,EACf,eAAe,EACf,qBAAqB,EACrB,yBAAyB,EACzB,oBAAoB,EACpB,eAAe,EACf,4BAA4B,EAC5B,wBAAwB,GAC3B,MAAM,sBAAsB,CAAC;AAG9B,OAAO,EACH,KAAK,EACL,eAAe,EACf,oBAAoB,EACpB,oBAAoB,EACpB,qBAAqB,EACrB,wBAAwB,EACxB,4BAA4B,EAC5B,uBAAuB,EACvB,iCAAiC,EACjC,UAAU,EACV,kBAAkB,EAClB,qBAAqB,EACrB,kBAAkB,EAClB,qBAAqB,EACrB,eAAe,EACf,uBAAuB,EACvB,iBAAiB,EACjB,kBAAkB,EAClB,mBAAmB,EACnB,uBAAuB,EACvB,qBAAqB,EACrB,kBAAkB,EAClB,kBAAkB,EAClB,kBAAkB,GACrB,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EACH,OAAO,EACP,uBAAuB,EACvB,wBAAwB,EACxB,eAAe,EACf,qBAAqB,EACrB,qBAAqB,EACrB,YAAY,EACZ,iBAAiB,EACjB,oBAAoB,GACvB,MAAM,sBAAsB,CAAC;AAG9B,OAAO,EACH,QAAQ,EACR,wBAAwB,EACxB,yBAAyB,EACzB,gBAAgB,EAChB,sBAAsB,EACtB,sBAAsB,EACtB,aAAa,EACb,kBAAkB,EAClB,qBAAqB,GACxB,MAAM,wBAAwB,CAAC;AAGhC,OAAO,EACH,OAAO,EACP,uBAAuB,EACvB,eAAe,EACf,oBAAoB,EACpB,YAAY,EACZ,uBAAuB,EACvB,iBAAiB,EACjB,wBAAwB,EACxB,wBAAwB,EACxB,oBAAoB,GACvB,MAAM,sBAAsB,CAAC;AAG9B,OAAO,EACH,IAAI,EACJ,0BAA0B,EAC1B,4BAA4B,EAC5B,mBAAmB,EACnB,uBAAuB,EACvB,SAAS,EACT,cAAc,EACd,qBAAqB,EACrB,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,GACnB,MAAM,gBAAgB,CAAC;AAGxB,OAAO,EACH,KAAK,EACL,eAAe,EACf,iBAAiB,EACjB,UAAU,EACV,mBAAmB,EACnB,aAAa,EACb,kBAAkB,EAClB,kBAAkB,EAClB,oBAAoB,EACpB,oBAAoB,EACpB,oBAAoB,EACpB,kBAAkB,GACrB,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EACH,EAAE,EACF,QAAQ,EACR,UAAU,EACV,cAAc,EACd,OAAO,EACP,YAAY,EACZ,YAAY,EACZ,mBAAmB,EACnB,eAAe,GAClB,MAAM,YAAY,CAAC;AAGpB,OAAO,EACH,UAAU,IAAI,gBAAgB,EAC9B,UAAU,IAAI,gBAAgB,EAC9B,mBAAmB,IAAI,yBAAyB,EAChD,mBAAmB,EACnB,eAAe,GAClB,MAAM,YAAY,CAAC"}

package/dist/index.js

@@ -0,0 +1,39 @@
+/**
+ * Copyright (c) 2025 Elara AI Pty Ltd
+ * Dual-licensed under AGPL-3.0 and commercial license. See LICENSE for details.
+ */
+/**
+ * East Data Science - ML and optimization platform functions for East.
+ *
+ * This package provides data science capabilities for East programs:
+ * - MADS: Derivative-free blackbox optimization (PyNomadBBO)
+ * - Optuna: Bayesian optimization with TPE sampler
+ * - SimAnneal: Discrete optimization with Simulated Annealing
+ *
+ * @packageDocumentation
+ */
+// MADS - Derivative-free optimization
+export { MADS, mads_optimize, MADSTypes, VectorType, ScalarObjectiveType, MADSBoundsType, MADSConstraintType, MADSDirectionType, MADSConfigType, MADSResultType, } from "./mads/mads.js";
+// Optuna - Bayesian optimization
+export { Optuna, optuna_optimize, OptunaTypes, ParamValueType, ParamSpaceKindType, ParamSpaceType, NamedParamType, OptimizationDirectionType, PrunerType, OptunaStudyConfigType, TrialResultType, StudyResultType, ObjectiveFunctionType, } from "./optuna/optuna.js";
+// SimAnneal - Discrete optimization
+export { SimAnneal, simanneal_optimize, simanneal_optimize_permutation, simanneal_optimize_subset, SimAnnealTypes, DiscreteStateType, EnergyFunctionType, MoveFunctionType, PermutationEnergyType, SubsetEnergyType, AnnealConfigType, AnnealResultType, } from "./simanneal/simanneal.js";
+// Sklearn - ML utilities
+export { Sklearn, sklearn_train_test_split, sklearn_standard_scaler_fit, sklearn_standard_scaler_transform, sklearn_min_max_scaler_fit, sklearn_min_max_scaler_transform, sklearn_metrics_regression, sklearn_metrics_classification, sklearn_regressor_chain_train, sklearn_regressor_chain_predict, SklearnTypes, SplitConfigType, SplitResultType, RegressionMetricsType, ClassificationMetricsType, SklearnModelBlobType, MultiTargetType, RegressorChainBaseConfigType, RegressorChainConfigType, } from "./sklearn/sklearn.js";
+// Scipy - Scientific computing
+export { Scipy, scipy_curve_fit, scipy_stats_describe, scipy_stats_pearsonr, scipy_stats_spearmanr, scipy_interpolate_1d_fit, scipy_interpolate_1d_predict, scipy_optimize_minimize, scipy_optimize_minimize_quadratic, ScipyTypes, OptimizeMethodType, InterpolationKindType, OptimizeConfigType, InterpolateConfigType, ParamBoundsType, CustomCurveFunctionType, CurveFunctionType, CurveFitConfigType, QuadraticConfigType, StatsDescribeResultType, CorrelationResultType, CurveFitResultType, OptimizeResultType, ScipyModelBlobType, } from "./scipy/scipy.js";
+// XGBoost - Gradient boosting
+export { XGBoost, xgboost_train_regressor, xgboost_train_classifier, xgboost_predict, xgboost_predict_class, xgboost_predict_proba, XGBoostTypes, XGBoostConfigType, XGBoostModelBlobType, } from "./xgboost/xgboost.js";
+// LightGBM - Fast gradient boosting
+export { LightGBM, lightgbm_train_regressor, lightgbm_train_classifier, lightgbm_predict, lightgbm_predict_class, lightgbm_predict_proba, LightGBMTypes, LightGBMConfigType, LightGBMModelBlobType, } from "./lightgbm/lightgbm.js";
+// NGBoost - Probabilistic gradient boosting
+export { NGBoost, ngboost_train_regressor, ngboost_predict, ngboost_predict_dist, NGBoostTypes, NGBoostDistributionType, NGBoostConfigType, NGBoostPredictConfigType, NGBoostPredictResultType, NGBoostModelBlobType, } from "./ngboost/ngboost.js";
+// SHAP - Model explainability
+export { Shap, shap_tree_explainer_create, shap_kernel_explainer_create, shap_compute_values, shap_feature_importance, ShapTypes, ShapResultType, FeatureImportanceType, ShapModelBlobType, AnyModelBlobType, StringVectorType, } from "./shap/shap.js";
+// Torch - PyTorch neural networks
+export { Torch, torch_mlp_train, torch_mlp_predict, TorchTypes, TorchActivationType, TorchLossType, TorchOptimizerType, TorchMLPConfigType, TorchTrainConfigType, TorchTrainResultType, TorchTrainOutputType, TorchModelBlobType, } from "./torch/torch.js";
+// GP - Gaussian Process regression
+export { GP, gp_train, gp_predict, gp_predict_std, GPTypes, GPKernelType, GPConfigType, GPPredictResultType, GPModelBlobType, } from "./gp/gp.js";
+// Shared types
+export { VectorType as SharedVectorType, MatrixType as SharedMatrixType, ScalarObjectiveType as SharedScalarObjectiveType, VectorObjectiveType, LabelVectorType, } from "./types.js";
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;;;GASG;AAEH,sCAAsC;AACtC,OAAO,EACH,IAAI,EACJ,aAAa,EACb,SAAS,EACT,UAAU,EACV,mBAAmB,EACnB,cAAc,EACd,kBAAkB,EAClB,iBAAiB,EACjB,cAAc,EACd,cAAc,GACjB,MAAM,gBAAgB,CAAC;AAExB,iCAAiC;AACjC,OAAO,EACH,MAAM,EACN,eAAe,EACf,WAAW,EACX,cAAc,EACd,kBAAkB,EAClB,cAAc,EACd,cAAc,EACd,yBAAyB,EACzB,UAAU,EACV,qBAAqB,EACrB,eAAe,EACf,eAAe,EACf,qBAAqB,GACxB,MAAM,oBAAoB,CAAC;AAE5B,oCAAoC;AACpC,OAAO,EACH,SAAS,EACT,kBAAkB,EAClB,8BAA8B,EAC9B,yBAAyB,EACzB,cAAc,EACd,iBAAiB,EACjB,kBAAkB,EAClB,gBAAgB,EAChB,qBAAqB,EACrB,gBAAgB,EAChB,gBAAgB,EAChB,gBAAgB,GACnB,MAAM,0BAA0B,CAAC;AAElC,yBAAyB;AACzB,OAAO,EACH,OAAO,EACP,wBAAwB,EACxB,2BAA2B,EAC3B,iCAAiC,EACjC,0BAA0B,EAC1B,gCAAgC,EAChC,0BAA0B,EAC1B,8BAA8B,EAC9B,6BAA6B,EAC7B,+BAA+B,EAC/B,YAAY,EACZ,eAAe,EACf,eAAe,EACf,qBAAqB,EACrB,yBAAyB,EACzB,oBAAoB,EACpB,eAAe,EACf,4BAA4B,EAC5B,wBAAwB,GAC3B,MAAM,sBAAsB,CAAC;AAE9B,+BAA+B;AAC/B,OAAO,EACH,KAAK,EACL,eAAe,EACf,oBAAoB,EACpB,oBAAoB,EACpB,qBAAqB,EACrB,wBAAwB,EACxB,4BAA4B,EAC5B,uBAAuB,EACvB,iCAAiC,EACjC,UAAU,EACV,kBAAkB,EAClB,qBAAqB,EACrB,kBAAkB,EAClB,qBAAqB,EACrB,eAAe,EACf,uBAAuB,EACvB,iBAAiB,EACjB,kBAAkB,EAClB,mBAAmB,EACnB,uBAAuB,EACvB,qBAAqB,EACrB,kBAAkB,EAClB,kBAAkB,EAClB,kBAAkB,GACrB,MAAM,kBAAkB,CAAC;AAE1B,8BAA8B;AAC9B,OAAO,EACH,OAAO,EACP,uBAAuB,EACvB,wBAAwB,EACxB,eAAe,EACf,qBAAqB,EACrB,qBAAqB,EACrB,YAAY,EACZ,iBAAiB,EACjB,oBAAoB,GACvB,MAAM,sBAAsB,CAAC;AAE9B,oCAAoC;AACpC,OAAO,EACH,QAAQ,EACR,wBAAwB,EACxB,yBAAyB,EACzB,gBAAgB,EAChB,sBAAsB,EACtB,sBAAsB,EACtB,aAAa,EACb,kBAAkB,EAClB,qBAAqB,GACxB,MAAM,wBAAwB,CAAC;AAEhC,4CAA4C;AAC5C,OAAO,EACH,OAAO,EACP,uBAAuB,EACvB,eAAe,EACf,oBAAoB,EACpB,YAAY,EACZ,uBAAuB,EACvB,iBAAiB,EACjB,wBAAwB,EACxB,wBAAwB,EACxB,oBAAoB,GACvB,MAAM,sBAAsB,CAAC;AAE9B,8BAA8B;AAC9B,OAAO,EACH,IAAI,EACJ,0BAA0B,EAC1B,4BAA4B,EAC5B,mBAAmB,EACnB,uBAAuB,EACvB,SAAS,EACT,cAAc,EACd,qBAAqB,EACrB,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,GACnB,MAAM,gBAAgB,CAAC;AAExB,kCAAkC;AAClC,OAAO,EACH,KAAK,EACL,eAAe,EACf,iBAAiB,EACjB,UAAU,EACV,mBAAmB,EACnB,aAAa,EACb,kBAAkB,EAClB,kBAAkB,EAClB,oBAAoB,EACpB,oBAAoB,EACpB,oBAAoB,EACpB,kBAAkB,GACrB,MAAM,kBAAkB,CAAC;AAE1B,mCAAmC;AACnC,OAAO,EACH,EAAE,EACF,QAAQ,EACR,UAAU,EACV,cAAc,EACd,OAAO,EACP,YAAY,EACZ,YAAY,EACZ,mBAAmB,EACnB,eAAe,GAClB,MAAM,YAAY,CAAC;AAEpB,eAAe;AACf,OAAO,EACH,UAAU,IAAI,gBAAgB,EAC9B,UAAU,IAAI,gBAAgB,EAC9B,mBAAmB,IAAI,yBAAyB,EAChD,mBAAmB,EACnB,eAAe,GAClB,MAAM,YAAY,CAAC"}