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

package/dist/sklearn/sklearn.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sklearn.d.ts","sourceRoot":"","sources":["../../src/sklearn/sklearn.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EAEH,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,WAAW,EACX,SAAS,EACT,QAAQ,EACR,SAAS,EACT,UAAU,EACb,MAAM,eAAe,CAAC;AAQvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAEtE,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAM3C;;GAEG;AACH,eAAO,MAAM,eAAe;IACxB,gEAAgE;;IAEhE,sCAAsC;;IAEtC,8DAA8D;;EAEhE,CAAC;AAMH;;GAEG;AACH,eAAO,MAAM,eAAe;IACxB,wBAAwB;;IAExB,oBAAoB;;IAEpB,sBAAsB;;IAEtB,kBAAkB;;EAEpB,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,qBAAqB;IAC9B,yBAAyB;;IAEzB,8BAA8B;;IAE9B,0BAA0B;;IAE1B,+CAA+C;;IAE/C,qCAAqC;;EAEvC,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,yBAAyB;IAClC,qBAAqB;;IAErB,mCAAmC;;IAEnC,gCAAgC;;IAEhC,kCAAkC;;EAEpC,CAAC;AAMH;;;;GAIG;AACH,eAAO,MAAM,oBAAoB;IAC7B,2BAA2B;;QAEvB,uBAAuB;;QAEvB,+BAA+B;;;IAGnC,yBAAyB;;QAErB,uBAAuB;;QAEvB,+BAA+B;;;IAGnC,2BAA2B;;QAEvB,mCAAmC;;QAEnC,+BAA+B;;QAE/B,+BAA+B;;QAE/B,+BAA+B;;;EAGrC,CAAC;AAMH;;;GAGG;AACH,eAAO,MAAM,4BAA4B;IACrC,wBAAwB;;;;;;;;;;;;;IAExB,yBAAyB;;;;;;;;;;;;;;IAEzB,wBAAwB;;;;;;;;;;;;IAExB,iCAAiC;;;;;;;;;;;;;;;EAEnC,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,wBAAwB;IACjC,4CAA4C;;QAd5C,wBAAwB;;;;;;;;;;;;;QAExB,yBAAyB;;;;;;;;;;;;;;QAEzB,wBAAwB;;;;;;;;;;;;QAExB,iCAAiC;;;;;;;;;;;;;;;;IAUjC,yEAAyE;;IAEzE,sCAAsC;;EAExC,CAAC;AAMH;;;;;;;GAOG;AACH,eAAO,MAAM,wBAAwB;IAxIjC,gEAAgE;;IAEhE,sCAAsC;;IAEtC,8DAA8D;;;IAY9D,wBAAwB;;IAExB,oBAAoB;;IAEpB,sBAAsB;;IAEtB,kBAAkB;;GAsHrB,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,2BAA2B;IApFpC,2BAA2B;;QAEvB,uBAAuB;;QAEvB,+BAA+B;;;IAGnC,yBAAyB;;QAErB,uBAAuB;;QAEvB,+BAA+B;;;IAGnC,2BAA2B;;QAEvB,mCAAmC;;QAEnC,+BAA+B;;QAE/B,+BAA+B;;QAE/B,+BAA+B;;;GAkEtC,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,iCAAiC;IAjG1C,2BAA2B;;QAEvB,uBAAuB;;QAEvB,+BAA+B;;;IAGnC,yBAAyB;;QAErB,uBAAuB;;QAEvB,+BAA+B;;;IAGnC,2BAA2B;;QAEvB,mCAAmC;;QAEnC,+BAA+B;;QAE/B,+BAA+B;;QAE/B,+BAA+B;;;sEA+EtC,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,0BAA0B;IA/GnC,2BAA2B;;QAEvB,uBAAuB;;QAEvB,+BAA+B;;;IAGnC,yBAAyB;;QAErB,uBAAuB;;QAEvB,+BAA+B;;;IAGnC,2BAA2B;;QAEvB,mCAAmC;;QAEnC,+BAA+B;;QAE/B,+BAA+B;;QAE/B,+BAA+B;;;GA6FtC,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,gCAAgC;IA5HzC,2BAA2B;;QAEvB,uBAAuB;;QAEvB,+BAA+B;;;IAGnC,yBAAyB;;QAErB,uBAAuB;;QAEvB,+BAA+B;;;IAGnC,2BAA2B;;QAEvB,mCAAmC;;QAEnC,+BAA+B;;QAE/B,+BAA+B;;QAE/B,+BAA+B;;;sEA0GtC,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,0BAA0B;IA7KnC,yBAAyB;;IAEzB,8BAA8B;;IAE9B,0BAA0B;;IAE1B,+CAA+C;;IAE/C,qCAAqC;;GAyKxC,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,8BAA8B;IA1KvC,qBAAqB;;IAErB,mCAAmC;;IAEnC,gCAAgC;;IAEhC,kCAAkC;;GAwKrC,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,eAAe,iCAAkC,CAAC;AAE/D;;;;;;;;;;GAUG;AACH,eAAO,MAAM,6BAA6B;IA1HtC,4CAA4C;;QAd5C,wBAAwB;;;;;;;;;;;;;QAExB,yBAAyB;;;;;;;;;;;;;;QAEzB,wBAAwB;;;;;;;;;;;;QAExB,iCAAiC;;;;;;;;;;;;;;;;IAUjC,yEAAyE;;IAEzE,sCAAsC;;;IAtDtC,2BAA2B;;QAEvB,uBAAuB;;QAEvB,+BAA+B;;;IAGnC,yBAAyB;;QAErB,uBAAuB;;QAEvB,+BAA+B;;;IAGnC,2BAA2B;;QAEvB,mCAAmC;;QAEnC,+BAA+B;;QAE/B,+BAA+B;;QAE/B,+BAA+B;;;GA0JtC,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,+BAA+B;IAzLxC,2BAA2B;;QAEvB,uBAAuB;;QAEvB,+BAA+B;;;IAGnC,yBAAyB;;QAErB,uBAAuB;;QAEvB,+BAA+B;;;IAGnC,2BAA2B;;QAEvB,mCAAmC;;QAEnC,+BAA+B;;QAE/B,+BAA+B;;QAE/B,+BAA+B;;;sEAuKtC,CAAC;AAMF;;GAEG;AACH,eAAO,MAAM,YAAY;IACrB,oCAAoC;;IAEpC,uCAAuC;;IAEvC,4CAA4C;;IAE5C,+BAA+B;;QA/Q/B,gEAAgE;;QAEhE,sCAAsC;;QAEtC,8DAA8D;;;IA6Q9D,wBAAwB;;QAjQxB,wBAAwB;;QAExB,oBAAoB;;QAEpB,sBAAsB;;QAEtB,kBAAkB;;;IA6PlB,8BAA8B;;QArP9B,yBAAyB;;QAEzB,8BAA8B;;QAE9B,0BAA0B;;QAE1B,+CAA+C;;QAE/C,qCAAqC;;;IA+OrC,kCAAkC;;QAvOlC,qBAAqB;;QAErB,mCAAmC;;QAEnC,gCAAgC;;QAEhC,kCAAkC;;;IAmOlC,yCAAyC;;QArNzC,2BAA2B;;YAEvB,uBAAuB;;YAEvB,+BAA+B;;;QAGnC,yBAAyB;;YAErB,uBAAuB;;YAEvB,+BAA+B;;;QAGnC,2BAA2B;;YAEvB,mCAAmC;;YAEnC,+BAA+B;;YAE/B,+BAA+B;;YAE/B,+BAA+B;;;;IAiMnC,+BAA+B;;IAE/B,gDAAgD;;QArLhD,wBAAwB;;;;;;;;;;;;;QAExB,yBAAyB;;;;;;;;;;;;;;QAEzB,wBAAwB;;;;;;;;;;;;QAExB,iCAAiC;;;;;;;;;;;;;;;;IAiLjC,iCAAiC;;QAzKjC,4CAA4C;;YAd5C,wBAAwB;;;;;;;;;;;;;YAExB,yBAAyB;;;;;;;;;;;;;;YAEzB,wBAAwB;;;;;;;;;;;;YAExB,iCAAiC;;;;;;;;;;;;;;;;QAUjC,yEAAyE;;QAEzE,sCAAsC;;;CAuKhC,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,eAAO,MAAM,OAAO;IAChB,+CAA+C;;QAxT/C,gEAAgE;;QAEhE,sCAAsC;;QAEtC,8DAA8D;;;QAY9D,wBAAwB;;QAExB,oBAAoB;;QAEpB,sBAAsB;;QAEtB,kBAAkB;;;IAoSlB,mCAAmC;;QAxPnC,2BAA2B;;YAEvB,uBAAuB;;YAEvB,+BAA+B;;;QAGnC,yBAAyB;;YAErB,uBAAuB;;YAEvB,+BAA+B;;;QAGnC,2BAA2B;;YAEvB,mCAAmC;;YAEnC,+BAA+B;;YAE/B,+BAA+B;;YAE/B,+BAA+B;;;;IAoOnC,iDAAiD;;QA1PjD,2BAA2B;;YAEvB,uBAAuB;;YAEvB,+BAA+B;;;QAGnC,yBAAyB;;YAErB,uBAAuB;;YAEvB,+BAA+B;;;QAGnC,2BAA2B;;YAEvB,mCAAmC;;YAEnC,+BAA+B;;YAE/B,+BAA+B;;YAE/B,+BAA+B;;;;IAsOnC,iCAAiC;;QA5PjC,2BAA2B;;YAEvB,uBAAuB;;YAEvB,+BAA+B;;;QAGnC,yBAAyB;;YAErB,uBAAuB;;YAEvB,+BAA+B;;;QAGnC,2BAA2B;;YAEvB,mCAAmC;;YAEnC,+BAA+B;;YAE/B,+BAA+B;;YAE/B,+BAA+B;;;;IAwOnC,+CAA+C;;QA9P/C,2BAA2B;;YAEvB,uBAAuB;;YAEvB,+BAA+B;;;QAGnC,yBAAyB;;YAErB,uBAAuB;;YAEvB,+BAA+B;;;QAGnC,2BAA2B;;YAEvB,mCAAmC;;YAEnC,+BAA+B;;YAE/B,+BAA+B;;YAE/B,+BAA+B;;;;IA0OnC,iCAAiC;;QApSjC,yBAAyB;;QAEzB,8BAA8B;;QAE9B,0BAA0B;;QAE1B,+CAA+C;;QAE/C,qCAAqC;;;IA8RrC,qCAAqC;;QAtRrC,qBAAqB;;QAErB,mCAAmC;;QAEnC,gCAAgC;;QAEhC,kCAAkC;;;IAkRlC,yDAAyD;;QAlNzD,4CAA4C;;YAd5C,wBAAwB;;;;;;;;;;;;;YAExB,yBAAyB;;;;;;;;;;;;;;YAEzB,wBAAwB;;;;;;;;;;;;YAExB,iCAAiC;;;;;;;;;;;;;;;;QAUjC,yEAAyE;;QAEzE,sCAAsC;;;QAtDtC,2BAA2B;;YAEvB,uBAAuB;;YAEvB,+BAA+B;;;QAGnC,yBAAyB;;YAErB,uBAAuB;;YAEvB,+BAA+B;;;QAGnC,2BAA2B;;YAEvB,mCAAmC;;YAEnC,+BAA+B;;YAE/B,+BAA+B;;YAE/B,+BAA+B;;;;IAgPnC,4CAA4C;;QAtQ5C,2BAA2B;;YAEvB,uBAAuB;;YAEvB,+BAA+B;;;QAGnC,yBAAyB;;YAErB,uBAAuB;;YAEvB,+BAA+B;;;QAGnC,2BAA2B;;YAEvB,mCAAmC;;YAEnC,+BAA+B;;YAE/B,+BAA+B;;YAE/B,+BAA+B;;;;IAkPnC,uBAAuB;;QAjEvB,oCAAoC;;QAEpC,uCAAuC;;QAEvC,4CAA4C;;QAE5C,+BAA+B;;YA/Q/B,gEAAgE;;YAEhE,sCAAsC;;YAEtC,8DAA8D;;;QA6Q9D,wBAAwB;;YAjQxB,wBAAwB;;YAExB,oBAAoB;;YAEpB,sBAAsB;;YAEtB,kBAAkB;;;QA6PlB,8BAA8B;;YArP9B,yBAAyB;;YAEzB,8BAA8B;;YAE9B,0BAA0B;;YAE1B,+CAA+C;;YAE/C,qCAAqC;;;QA+OrC,kCAAkC;;YAvOlC,qBAAqB;;YAErB,mCAAmC;;YAEnC,gCAAgC;;YAEhC,kCAAkC;;;QAmOlC,yCAAyC;;YArNzC,2BAA2B;;gBAEvB,uBAAuB;;gBAEvB,+BAA+B;;;YAGnC,yBAAyB;;gBAErB,uBAAuB;;gBAEvB,+BAA+B;;;YAGnC,2BAA2B;;gBAEvB,mCAAmC;;gBAEnC,+BAA+B;;gBAE/B,+BAA+B;;gBAE/B,+BAA+B;;;;QAiMnC,+BAA+B;;QAE/B,gDAAgD;;YArLhD,wBAAwB;;;;;;;;;;;;;YAExB,yBAAyB;;;;;;;;;;;;;;YAEzB,wBAAwB;;;;;;;;;;;;YAExB,iCAAiC;;;;;;;;;;;;;;;;QAiLjC,iCAAiC;;YAzKjC,4CAA4C;;gBAd5C,wBAAwB;;;;;;;;;;;;;gBAExB,yBAAyB;;;;;;;;;;;;;;gBAEzB,wBAAwB;;;;;;;;;;;;gBAExB,iCAAiC;;;;;;;;;;;;;;;;YAUjC,yEAAyE;;YAEzE,sCAAsC;;;;CAoNhC,CAAC"}

package/dist/sklearn/sklearn.js

@@ -0,0 +1,307 @@
+/**
+ * Copyright (c) 2025 Elara AI Pty Ltd
+ * Dual-licensed under AGPL-3.0 and commercial license. See LICENSE for details.
+ */
+/**
+ * Scikit-learn platform functions for East.
+ *
+ * Provides core machine learning utilities: preprocessing, model selection, and metrics.
+ * Uses ONNX for model serialization to enable portable inference.
+ *
+ * @packageDocumentation
+ */
+import { East, StructType, VariantType, OptionType, IntegerType, BooleanType, FloatType, BlobType, ArrayType, StringType, } from "@elaraai/east";
+import { VectorType, MatrixType, LabelVectorType } from "../types.js";
+import { XGBoostConfigType } from "../xgboost/xgboost.js";
+import { LightGBMConfigType } from "../lightgbm/lightgbm.js";
+import { NGBoostConfigType } from "../ngboost/ngboost.js";
+import { GPConfigType } from "../gp/gp.js";
+// Re-export shared types for convenience
+export { VectorType, MatrixType, LabelVectorType } from "../types.js";
+// Re-export config types used in RegressorChain
+export { XGBoostConfigType } from "../xgboost/xgboost.js";
+export { LightGBMConfigType } from "../lightgbm/lightgbm.js";
+export { NGBoostConfigType } from "../ngboost/ngboost.js";
+export { GPConfigType } from "../gp/gp.js";
+// ============================================================================
+// Config Types
+// ============================================================================
+/**
+ * Configuration for train/test split.
+ */
+export const SplitConfigType = StructType({
+    /** Proportion of data to include in test split (default 0.2) */
+    test_size: OptionType(FloatType),
+    /** Random seed for reproducibility */
+    random_state: OptionType(IntegerType),
+    /** Whether to shuffle data before splitting (default true) */
+    shuffle: OptionType(BooleanType),
+});
+// ============================================================================
+// Result Types
+// ============================================================================
+/**
+ * Result of train/test split.
+ */
+export const SplitResultType = StructType({
+    /** Training features */
+    X_train: MatrixType,
+    /** Test features */
+    X_test: MatrixType,
+    /** Training labels */
+    y_train: VectorType,
+    /** Test labels */
+    y_test: VectorType,
+});
+/**
+ * Regression metrics result.
+ */
+export const RegressionMetricsType = StructType({
+    /** Mean Squared Error */
+    mse: FloatType,
+    /** Root Mean Squared Error */
+    rmse: FloatType,
+    /** Mean Absolute Error */
+    mae: FloatType,
+    /** R-squared (coefficient of determination) */
+    r2: FloatType,
+    /** Mean Absolute Percentage Error */
+    mape: FloatType,
+});
+/**
+ * Classification metrics result.
+ */
+export const ClassificationMetricsType = StructType({
+    /** Accuracy score */
+    accuracy: FloatType,
+    /** Precision (weighted average) */
+    precision: FloatType,
+    /** Recall (weighted average) */
+    recall: FloatType,
+    /** F1 score (weighted average) */
+    f1: FloatType,
+});
+// ============================================================================
+// Model Blob Types
+// ============================================================================
+/**
+ * Model blob type for serialized sklearn models.
+ *
+ * Each model type has its own variant case containing ONNX bytes and metadata.
+ */
+export const SklearnModelBlobType = VariantType({
+    /** StandardScaler model */
+    standard_scaler: StructType({
+        /** ONNX model bytes */
+        onnx: BlobType,
+        /** Number of input features */
+        n_features: IntegerType,
+    }),
+    /** MinMaxScaler model */
+    min_max_scaler: StructType({
+        /** ONNX model bytes */
+        onnx: BlobType,
+        /** Number of input features */
+        n_features: IntegerType,
+    }),
+    /** RegressorChain model */
+    regressor_chain: StructType({
+        /** Cloudpickle serialized chain */
+        data: BlobType,
+        /** Number of input features */
+        n_features: IntegerType,
+        /** Number of target outputs */
+        n_targets: IntegerType,
+        /** Base estimator type name */
+        base_estimator_type: StringType,
+    }),
+});
+// ============================================================================
+// RegressorChain Types
+// ============================================================================
+/**
+ * Base estimator configuration for RegressorChain.
+ * Variant carries both the estimator type AND its configuration.
+ */
+export const RegressorChainBaseConfigType = VariantType({
+    /** XGBoost regressor */
+    xgboost: XGBoostConfigType,
+    /** LightGBM regressor */
+    lightgbm: LightGBMConfigType,
+    /** NGBoost regressor */
+    ngboost: NGBoostConfigType,
+    /** Gaussian Process regressor */
+    gp: GPConfigType,
+});
+/**
+ * Configuration for RegressorChain.
+ */
+export const RegressorChainConfigType = StructType({
+    /** Base estimator with its configuration */
+    base_estimator: RegressorChainBaseConfigType,
+    /** Chain order (indices of targets). None = natural order [0,1,2,...] */
+    order: OptionType(ArrayType(IntegerType)),
+    /** Random seed for reproducibility */
+    random_state: OptionType(IntegerType),
+});
+// ============================================================================
+// Platform Functions
+// ============================================================================
+/**
+ * Split arrays into train and test subsets.
+ *
+ * @param X - Feature matrix
+ * @param y - Target vector
+ * @param config - Split configuration
+ * @returns Split result with X_train, X_test, y_train, y_test
+ */
+export const sklearn_train_test_split = East.platform("sklearn_train_test_split", [MatrixType, VectorType, SplitConfigType], SplitResultType);
+/**
+ * Fit a StandardScaler to training data.
+ *
+ * Standardizes features by removing the mean and scaling to unit variance.
+ *
+ * @param X - Training feature matrix
+ * @returns Model blob containing fitted scaler
+ */
+export const sklearn_standard_scaler_fit = East.platform("sklearn_standard_scaler_fit", [MatrixType], SklearnModelBlobType);
+/**
+ * Transform data using a fitted StandardScaler.
+ *
+ * @param model - Fitted scaler model blob
+ * @param X - Feature matrix to transform
+ * @returns Transformed feature matrix
+ */
+export const sklearn_standard_scaler_transform = East.platform("sklearn_standard_scaler_transform", [SklearnModelBlobType, MatrixType], MatrixType);
+/**
+ * Fit a MinMaxScaler to training data.
+ *
+ * Scales features to a given range (default [0, 1]).
+ *
+ * @param X - Training feature matrix
+ * @returns Model blob containing fitted scaler
+ */
+export const sklearn_min_max_scaler_fit = East.platform("sklearn_min_max_scaler_fit", [MatrixType], SklearnModelBlobType);
+/**
+ * Transform data using a fitted MinMaxScaler.
+ *
+ * @param model - Fitted scaler model blob
+ * @param X - Feature matrix to transform
+ * @returns Transformed feature matrix
+ */
+export const sklearn_min_max_scaler_transform = East.platform("sklearn_min_max_scaler_transform", [SklearnModelBlobType, MatrixType], MatrixType);
+/**
+ * Compute regression metrics.
+ *
+ * @param y_true - True target values
+ * @param y_pred - Predicted target values
+ * @returns Regression metrics (MSE, RMSE, MAE, R2, MAPE)
+ */
+export const sklearn_metrics_regression = East.platform("sklearn_metrics_regression", [VectorType, VectorType], RegressionMetricsType);
+/**
+ * Compute classification metrics.
+ *
+ * @param y_true - True class labels
+ * @param y_pred - Predicted class labels
+ * @returns Classification metrics (accuracy, precision, recall, F1)
+ */
+export const sklearn_metrics_classification = East.platform("sklearn_metrics_classification", [LabelVectorType, LabelVectorType], ClassificationMetricsType);
+/**
+ * Multi-target output type for RegressorChain.
+ */
+export const MultiTargetType = ArrayType(ArrayType(FloatType));
+/**
+ * Train a RegressorChain for multi-target regression.
+ *
+ * Each model in the chain uses previous targets as additional features,
+ * enabling modeling of dependencies between targets.
+ *
+ * @param X - Feature matrix
+ * @param Y - Target matrix (rows=samples, cols=targets)
+ * @param config - Chain configuration
+ * @returns Model blob containing fitted chain
+ */
+export const sklearn_regressor_chain_train = East.platform("sklearn_regressor_chain_train", [MatrixType, MultiTargetType, RegressorChainConfigType], SklearnModelBlobType);
+/**
+ * Predict using a fitted RegressorChain.
+ *
+ * @param model - Fitted chain model blob
+ * @param X - Feature matrix to predict
+ * @returns Predicted target matrix
+ */
+export const sklearn_regressor_chain_predict = East.platform("sklearn_regressor_chain_predict", [SklearnModelBlobType, MatrixType], MultiTargetType);
+// ============================================================================
+// Grouped Export
+// ============================================================================
+/**
+ * Type definitions for sklearn functions.
+ */
+export const SklearnTypes = {
+    /** Vector type (array of floats) */
+    VectorType,
+    /** Matrix type (2D array of floats) */
+    MatrixType,
+    /** Label vector type (array of integers) */
+    LabelVectorType,
+    /** Split configuration type */
+    SplitConfigType,
+    /** Split result type */
+    SplitResultType,
+    /** Regression metrics type */
+    RegressionMetricsType,
+    /** Classification metrics type */
+    ClassificationMetricsType,
+    /** Model blob type for sklearn models */
+    ModelBlobType: SklearnModelBlobType,
+    /** Multi-target output type */
+    MultiTargetType,
+    /** RegressorChain base estimator config type */
+    RegressorChainBaseConfigType,
+    /** RegressorChain config type */
+    RegressorChainConfigType,
+};
+/**
+ * Scikit-learn machine learning utilities.
+ *
+ * Provides preprocessing, model selection, and metrics for ML workflows.
+ *
+ * @example
+ * ```ts
+ * import { East, variant } from "@elaraai/east";
+ * import { Sklearn } from "@elaraai/east-py-datascience";
+ *
+ * const pipeline = East.function([], Sklearn.Types.SplitResultType, $ => {
+ *     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({
+ *         test_size: variant('some', 0.25),
+ *         random_state: variant('some', 42n),
+ *         shuffle: variant('some', true),
+ *     });
+ *     return $.return(Sklearn.trainTestSplit(X, y, config));
+ * });
+ * ```
+ */
+export const Sklearn = {
+    /** Split arrays into train and test subsets */
+    trainTestSplit: sklearn_train_test_split,
+    /** Fit a StandardScaler to data */
+    standardScalerFit: sklearn_standard_scaler_fit,
+    /** Transform data using fitted StandardScaler */
+    standardScalerTransform: sklearn_standard_scaler_transform,
+    /** Fit a MinMaxScaler to data */
+    minMaxScalerFit: sklearn_min_max_scaler_fit,
+    /** Transform data using fitted MinMaxScaler */
+    minMaxScalerTransform: sklearn_min_max_scaler_transform,
+    /** Compute regression metrics */
+    metricsRegression: sklearn_metrics_regression,
+    /** Compute classification metrics */
+    metricsClassification: sklearn_metrics_classification,
+    /** Train a RegressorChain for multi-target regression */
+    regressorChainTrain: sklearn_regressor_chain_train,
+    /** Predict using a fitted RegressorChain */
+    regressorChainPredict: sklearn_regressor_chain_predict,
+    /** Type definitions */
+    Types: SklearnTypes,
+};
+//# sourceMappingURL=sklearn.js.map

package/dist/sklearn/sklearn.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sklearn.js","sourceRoot":"","sources":["../../src/sklearn/sklearn.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EACH,IAAI,EACJ,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,WAAW,EACX,SAAS,EACT,QAAQ,EACR,SAAS,EACT,UAAU,GACb,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AACtE,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,yCAAyC;AACzC,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AACtE,gDAAgD;AAChD,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,iBAAiB,EAAE,MAAM,uBAAuB,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAE3C,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,UAAU,CAAC;IACtC,gEAAgE;IAChE,SAAS,EAAE,UAAU,CAAC,SAAS,CAAC;IAChC,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,8DAA8D;IAC9D,OAAO,EAAE,UAAU,CAAC,WAAW,CAAC;CACnC,CAAC,CAAC;AAEH,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,UAAU,CAAC;IACtC,wBAAwB;IACxB,OAAO,EAAE,UAAU;IACnB,oBAAoB;IACpB,MAAM,EAAE,UAAU;IAClB,sBAAsB;IACtB,OAAO,EAAE,UAAU;IACnB,kBAAkB;IAClB,MAAM,EAAE,UAAU;CACrB,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,UAAU,CAAC;IAC5C,yBAAyB;IACzB,GAAG,EAAE,SAAS;IACd,8BAA8B;IAC9B,IAAI,EAAE,SAAS;IACf,0BAA0B;IAC1B,GAAG,EAAE,SAAS;IACd,+CAA+C;IAC/C,EAAE,EAAE,SAAS;IACb,qCAAqC;IACrC,IAAI,EAAE,SAAS;CAClB,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,UAAU,CAAC;IAChD,qBAAqB;IACrB,QAAQ,EAAE,SAAS;IACnB,mCAAmC;IACnC,SAAS,EAAE,SAAS;IACpB,gCAAgC;IAChC,MAAM,EAAE,SAAS;IACjB,kCAAkC;IAClC,EAAE,EAAE,SAAS;CAChB,CAAC,CAAC;AAEH,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;GAIG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,WAAW,CAAC;IAC5C,2BAA2B;IAC3B,eAAe,EAAE,UAAU,CAAC;QACxB,uBAAuB;QACvB,IAAI,EAAE,QAAQ;QACd,+BAA+B;QAC/B,UAAU,EAAE,WAAW;KAC1B,CAAC;IACF,yBAAyB;IACzB,cAAc,EAAE,UAAU,CAAC;QACvB,uBAAuB;QACvB,IAAI,EAAE,QAAQ;QACd,+BAA+B;QAC/B,UAAU,EAAE,WAAW;KAC1B,CAAC;IACF,2BAA2B;IAC3B,eAAe,EAAE,UAAU,CAAC;QACxB,mCAAmC;QACnC,IAAI,EAAE,QAAQ;QACd,+BAA+B;QAC/B,UAAU,EAAE,WAAW;QACvB,+BAA+B;QAC/B,SAAS,EAAE,WAAW;QACtB,+BAA+B;QAC/B,mBAAmB,EAAE,UAAU;KAClC,CAAC;CACL,CAAC,CAAC;AAEH,+EAA+E;AAC/E,uBAAuB;AACvB,+EAA+E;AAE/E;;;GAGG;AACH,MAAM,CAAC,MAAM,4BAA4B,GAAG,WAAW,CAAC;IACpD,wBAAwB;IACxB,OAAO,EAAE,iBAAiB;IAC1B,yBAAyB;IACzB,QAAQ,EAAE,kBAAkB;IAC5B,wBAAwB;IACxB,OAAO,EAAE,iBAAiB;IAC1B,iCAAiC;IACjC,EAAE,EAAE,YAAY;CACnB,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,wBAAwB,GAAG,UAAU,CAAC;IAC/C,4CAA4C;IAC5C,cAAc,EAAE,4BAA4B;IAC5C,yEAAyE;IACzE,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;IACzC,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;CACxC,CAAC,CAAC;AAEH,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAE/E;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,wBAAwB,GAAG,IAAI,CAAC,QAAQ,CACjD,0BAA0B,EAC1B,CAAC,UAAU,EAAE,UAAU,EAAE,eAAe,CAAC,EACzC,eAAe,CAClB,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,2BAA2B,GAAG,IAAI,CAAC,QAAQ,CACpD,6BAA6B,EAC7B,CAAC,UAAU,CAAC,EACZ,oBAAoB,CACvB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,iCAAiC,GAAG,IAAI,CAAC,QAAQ,CAC1D,mCAAmC,EACnC,CAAC,oBAAoB,EAAE,UAAU,CAAC,EAClC,UAAU,CACb,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,IAAI,CAAC,QAAQ,CACnD,4BAA4B,EAC5B,CAAC,UAAU,CAAC,EACZ,oBAAoB,CACvB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,gCAAgC,GAAG,IAAI,CAAC,QAAQ,CACzD,kCAAkC,EAClC,CAAC,oBAAoB,EAAE,UAAU,CAAC,EAClC,UAAU,CACb,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,IAAI,CAAC,QAAQ,CACnD,4BAA4B,EAC5B,CAAC,UAAU,EAAE,UAAU,CAAC,EACxB,qBAAqB,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,8BAA8B,GAAG,IAAI,CAAC,QAAQ,CACvD,gCAAgC,EAChC,CAAC,eAAe,EAAE,eAAe,CAAC,EAClC,yBAAyB,CAC5B,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC;AAE/D;;;;;;;;;;GAUG;AACH,MAAM,CAAC,MAAM,6BAA6B,GAAG,IAAI,CAAC,QAAQ,CACtD,+BAA+B,EAC/B,CAAC,UAAU,EAAE,eAAe,EAAE,wBAAwB,CAAC,EACvD,oBAAoB,CACvB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,+BAA+B,GAAG,IAAI,CAAC,QAAQ,CACxD,iCAAiC,EACjC,CAAC,oBAAoB,EAAE,UAAU,CAAC,EAClC,eAAe,CAClB,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,+BAA+B;IAC/B,eAAe;IACf,wBAAwB;IACxB,eAAe;IACf,8BAA8B;IAC9B,qBAAqB;IACrB,kCAAkC;IAClC,yBAAyB;IACzB,yCAAyC;IACzC,aAAa,EAAE,oBAAoB;IACnC,+BAA+B;IAC/B,eAAe;IACf,gDAAgD;IAChD,4BAA4B;IAC5B,iCAAiC;IACjC,wBAAwB;CAClB,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,CAAC,MAAM,OAAO,GAAG;IACnB,+CAA+C;IAC/C,cAAc,EAAE,wBAAwB;IACxC,mCAAmC;IACnC,iBAAiB,EAAE,2BAA2B;IAC9C,iDAAiD;IACjD,uBAAuB,EAAE,iCAAiC;IAC1D,iCAAiC;IACjC,eAAe,EAAE,0BAA0B;IAC3C,+CAA+C;IAC/C,qBAAqB,EAAE,gCAAgC;IACvD,iCAAiC;IACjC,iBAAiB,EAAE,0BAA0B;IAC7C,qCAAqC;IACrC,qBAAqB,EAAE,8BAA8B;IACrD,yDAAyD;IACzD,mBAAmB,EAAE,6BAA6B;IAClD,4CAA4C;IAC5C,qBAAqB,EAAE,+BAA+B;IACtD,uBAAuB;IACvB,KAAK,EAAE,YAAY;CACb,CAAC"}