@elaraai/east-py-datascience 0.0.2-beta.55 → 0.0.2-beta.56

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (10) hide show
  1. package/dist/index.d.ts +1 -1
  2. package/dist/index.d.ts.map +1 -1
  3. package/dist/index.js +1 -1
  4. package/dist/index.js.map +1 -1
  5. package/dist/sklearn/sklearn.d.ts +131 -246
  6. package/dist/sklearn/sklearn.d.ts.map +1 -1
  7. package/dist/sklearn/sklearn.js +51 -74
  8. package/dist/sklearn/sklearn.js.map +1 -1
  9. package/dist/tsconfig.tsbuildinfo +1 -1
  10. package/package.json +1 -1
package/dist/index.d.ts CHANGED
@@ -15,7 +15,7 @@
15
15
  export { MADS, mads_optimize, MADSTypes, VectorType, ScalarObjectiveType, MADSBoundsType, MADSConstraintType, MADSDirectionType, MADSConfigType, MADSResultType, } from "./mads/mads.js";
16
16
  export { Optuna, optuna_optimize, OptunaTypes, ParamValueType, ParamSpaceKindType, ParamSpaceType, NamedParamType, OptimizationDirectionType, PrunerType, OptunaStudyConfigType, TrialResultType, StudyResultType, ObjectiveFunctionType, } from "./optuna/optuna.js";
17
17
  export { SimAnneal, simanneal_optimize, simanneal_optimize_permutation, simanneal_optimize_subset, SimAnnealTypes, DiscreteStateType, EnergyFunctionType, MoveFunctionType, PermutationEnergyType, SubsetEnergyType, AnnealConfigType, AnnealResultType, } from "./simanneal/simanneal.js";
18
- export { Sklearn, sklearn_train_test_split, sklearn_train_val_test_split, sklearn_standard_scaler_fit, sklearn_standard_scaler_transform, sklearn_min_max_scaler_fit, sklearn_min_max_scaler_transform, sklearn_compute_metrics, sklearn_compute_metrics_multi, sklearn_compute_classification_metrics, sklearn_compute_classification_metrics_multi, sklearn_regressor_chain_train, sklearn_regressor_chain_predict, SklearnTypes, SplitConfigType, SplitResultType, ThreeWaySplitConfigType, ThreeWaySplitResultType, SklearnModelBlobType, RegressorChainBaseConfigType, RegressorChainConfigType, RegressionMetricType, MetricResultType, MetricsResultType, MetricAggregationType, MultiMetricsConfigType, MultiMetricResultType, MultiMetricsResultType, ClassificationMetricType, ClassificationAverageType, ClassificationMetricsConfigType, ClassificationMetricResultType, ClassificationMetricResultsType, MultiClassificationConfigType, MultiClassificationMetricResultType, MultiClassificationMetricResultsType, } from "./sklearn/sklearn.js";
18
+ export { Sklearn, sklearn_split, sklearn_standard_scaler_fit, sklearn_standard_scaler_transform, sklearn_min_max_scaler_fit, sklearn_min_max_scaler_transform, sklearn_compute_metrics, sklearn_compute_metrics_multi, sklearn_compute_classification_metrics, sklearn_compute_classification_metrics_multi, sklearn_regressor_chain_train, sklearn_regressor_chain_predict, SklearnTypes, SplitConfigType, SplitResultType, SklearnModelBlobType, RegressorChainBaseConfigType, RegressorChainConfigType, RegressionMetricType, MetricResultType, MetricsResultType, MetricAggregationType, MultiMetricsConfigType, MultiMetricResultType, MultiMetricsResultType, ClassificationMetricType, ClassificationAverageType, ClassificationMetricsConfigType, ClassificationMetricResultType, ClassificationMetricResultsType, MultiClassificationConfigType, MultiClassificationMetricResultType, MultiClassificationMetricResultsType, } from "./sklearn/sklearn.js";
19
19
  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";
20
20
  export { XGBoost, xgboost_train_regressor, xgboost_train_classifier, xgboost_train_quantile, xgboost_predict, xgboost_predict_class, xgboost_predict_proba, xgboost_predict_quantile, XGBoostTypes, XGBoostConfigType, XGBoostQuantileConfigType, XGBoostQuantilePredictResultType, XGBoostModelBlobType, } from "./xgboost/xgboost.js";
21
21
  export { LightGBM, lightgbm_train_regressor, lightgbm_train_classifier, lightgbm_predict, lightgbm_predict_class, lightgbm_predict_proba, LightGBMTypes, LightGBMConfigType, LightGBMModelBlobType, } from "./lightgbm/lightgbm.js";
package/dist/index.d.ts.map CHANGED
@@ -1 +1 @@
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,4BAA4B,EAC5B,2BAA2B,EAC3B,iCAAiC,EACjC,0BAA0B,EAC1B,gCAAgC,EAChC,uBAAuB,EACvB,6BAA6B,EAC7B,sCAAsC,EACtC,4CAA4C,EAC5C,6BAA6B,EAC7B,+BAA+B,EAC/B,YAAY,EACZ,eAAe,EACf,eAAe,EACf,uBAAuB,EACvB,uBAAuB,EACvB,oBAAoB,EACpB,4BAA4B,EAC5B,wBAAwB,EAExB,oBAAoB,EACpB,gBAAgB,EAChB,iBAAiB,EACjB,qBAAqB,EACrB,sBAAsB,EACtB,qBAAqB,EACrB,sBAAsB,EACtB,wBAAwB,EACxB,yBAAyB,EACzB,+BAA+B,EAC/B,8BAA8B,EAC9B,+BAA+B,EAC/B,6BAA6B,EAC7B,mCAAmC,EACnC,oCAAoC,GACvC,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,sBAAsB,EACtB,eAAe,EACf,qBAAqB,EACrB,qBAAqB,EACrB,wBAAwB,EACxB,YAAY,EACZ,iBAAiB,EACjB,yBAAyB,EACzB,gCAAgC,EAChC,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,SAAS,EACT,eAAe,EACf,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,EAChB,cAAc,EACd,mBAAmB,EACnB,yBAAyB,EACzB,0BAA0B,EAC1B,mBAAmB,EACnB,mBAAmB,EACnB,sBAAsB,EACtB,gBAAgB,GACnB,MAAM,0BAA0B,CAAC;AAGlC,OAAO,EACH,KAAK,EACL,+BAA+B,EAC/B,eAAe,EACf,sBAAsB,EACtB,gCAAgC,EAChC,iBAAiB,EACjB,UAAU,EACV,mBAAmB,EACnB,aAAa,EACb,eAAe,EACf,kBAAkB,EAClB,wBAAwB,EACxB,kBAAkB,EAClB,yBAAyB,EACzB,sBAAsB,EACtB,uBAAuB,EACvB,kBAAkB,EAClB,uBAAuB,GAC1B,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EACH,IAAI,EACJ,aAAa,EACb,SAAS,EACT,4BAA4B,EAC5B,wBAAwB,EACxB,uBAAuB,EACvB,uBAAuB,EACvB,qBAAqB,EACrB,iBAAiB,EACjB,cAAc,EACd,cAAc,GACjB,MAAM,gBAAgB,CAAC;AAGxB,OAAO,EACH,UAAU,IAAI,gBAAgB,EAC9B,UAAU,IAAI,gBAAgB,EAC9B,mBAAmB,IAAI,yBAAyB,EAChD,mBAAmB,EACnB,eAAe,GAClB,MAAM,YAAY,CAAC"}
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,aAAa,EACb,2BAA2B,EAC3B,iCAAiC,EACjC,0BAA0B,EAC1B,gCAAgC,EAChC,uBAAuB,EACvB,6BAA6B,EAC7B,sCAAsC,EACtC,4CAA4C,EAC5C,6BAA6B,EAC7B,+BAA+B,EAC/B,YAAY,EACZ,eAAe,EACf,eAAe,EACf,oBAAoB,EACpB,4BAA4B,EAC5B,wBAAwB,EAExB,oBAAoB,EACpB,gBAAgB,EAChB,iBAAiB,EACjB,qBAAqB,EACrB,sBAAsB,EACtB,qBAAqB,EACrB,sBAAsB,EACtB,wBAAwB,EACxB,yBAAyB,EACzB,+BAA+B,EAC/B,8BAA8B,EAC9B,+BAA+B,EAC/B,6BAA6B,EAC7B,mCAAmC,EACnC,oCAAoC,GACvC,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,sBAAsB,EACtB,eAAe,EACf,qBAAqB,EACrB,qBAAqB,EACrB,wBAAwB,EACxB,YAAY,EACZ,iBAAiB,EACjB,yBAAyB,EACzB,gCAAgC,EAChC,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,SAAS,EACT,eAAe,EACf,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,EAChB,cAAc,EACd,mBAAmB,EACnB,yBAAyB,EACzB,0BAA0B,EAC1B,mBAAmB,EACnB,mBAAmB,EACnB,sBAAsB,EACtB,gBAAgB,GACnB,MAAM,0BAA0B,CAAC;AAGlC,OAAO,EACH,KAAK,EACL,+BAA+B,EAC/B,eAAe,EACf,sBAAsB,EACtB,gCAAgC,EAChC,iBAAiB,EACjB,UAAU,EACV,mBAAmB,EACnB,aAAa,EACb,eAAe,EACf,kBAAkB,EAClB,wBAAwB,EACxB,kBAAkB,EAClB,yBAAyB,EACzB,sBAAsB,EACtB,uBAAuB,EACvB,kBAAkB,EAClB,uBAAuB,GAC1B,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EACH,IAAI,EACJ,aAAa,EACb,SAAS,EACT,4BAA4B,EAC5B,wBAAwB,EACxB,uBAAuB,EACvB,uBAAuB,EACvB,qBAAqB,EACrB,iBAAiB,EACjB,cAAc,EACd,cAAc,GACjB,MAAM,gBAAgB,CAAC;AAGxB,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 CHANGED
@@ -19,7 +19,7 @@ export { Optuna, optuna_optimize, OptunaTypes, ParamValueType, ParamSpaceKindTyp
19
19
  // SimAnneal - Discrete optimization
20
20
  export { SimAnneal, simanneal_optimize, simanneal_optimize_permutation, simanneal_optimize_subset, SimAnnealTypes, DiscreteStateType, EnergyFunctionType, MoveFunctionType, PermutationEnergyType, SubsetEnergyType, AnnealConfigType, AnnealResultType, } from "./simanneal/simanneal.js";
21
21
  // Sklearn - ML utilities
22
- export { Sklearn, sklearn_train_test_split, sklearn_train_val_test_split, sklearn_standard_scaler_fit, sklearn_standard_scaler_transform, sklearn_min_max_scaler_fit, sklearn_min_max_scaler_transform, sklearn_compute_metrics, sklearn_compute_metrics_multi, sklearn_compute_classification_metrics, sklearn_compute_classification_metrics_multi, sklearn_regressor_chain_train, sklearn_regressor_chain_predict, SklearnTypes, SplitConfigType, SplitResultType, ThreeWaySplitConfigType, ThreeWaySplitResultType, SklearnModelBlobType, RegressorChainBaseConfigType, RegressorChainConfigType,
22
+ export { Sklearn, sklearn_split, sklearn_standard_scaler_fit, sklearn_standard_scaler_transform, sklearn_min_max_scaler_fit, sklearn_min_max_scaler_transform, sklearn_compute_metrics, sklearn_compute_metrics_multi, sklearn_compute_classification_metrics, sklearn_compute_classification_metrics_multi, sklearn_regressor_chain_train, sklearn_regressor_chain_predict, SklearnTypes, SplitConfigType, SplitResultType, SklearnModelBlobType, RegressorChainBaseConfigType, RegressorChainConfigType,
23
23
  // Flexible metrics types
24
24
  RegressionMetricType, MetricResultType, MetricsResultType, MetricAggregationType, MultiMetricsConfigType, MultiMetricResultType, MultiMetricsResultType, ClassificationMetricType, ClassificationAverageType, ClassificationMetricsConfigType, ClassificationMetricResultType, ClassificationMetricResultsType, MultiClassificationConfigType, MultiClassificationMetricResultType, MultiClassificationMetricResultsType, } from "./sklearn/sklearn.js";
25
25
  // Scipy - Scientific computing
package/dist/index.js.map CHANGED
@@ -1 +1 @@
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,4BAA4B,EAC5B,2BAA2B,EAC3B,iCAAiC,EACjC,0BAA0B,EAC1B,gCAAgC,EAChC,uBAAuB,EACvB,6BAA6B,EAC7B,sCAAsC,EACtC,4CAA4C,EAC5C,6BAA6B,EAC7B,+BAA+B,EAC/B,YAAY,EACZ,eAAe,EACf,eAAe,EACf,uBAAuB,EACvB,uBAAuB,EACvB,oBAAoB,EACpB,4BAA4B,EAC5B,wBAAwB;AACxB,yBAAyB;AACzB,oBAAoB,EACpB,gBAAgB,EAChB,iBAAiB,EACjB,qBAAqB,EACrB,sBAAsB,EACtB,qBAAqB,EACrB,sBAAsB,EACtB,wBAAwB,EACxB,yBAAyB,EACzB,+BAA+B,EAC/B,8BAA8B,EAC9B,+BAA+B,EAC/B,6BAA6B,EAC7B,mCAAmC,EACnC,oCAAoC,GACvC,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,sBAAsB,EACtB,eAAe,EACf,qBAAqB,EACrB,qBAAqB,EACrB,wBAAwB,EACxB,YAAY,EACZ,iBAAiB,EACjB,yBAAyB,EACzB,gCAAgC,EAChC,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,gDAAgD;AAChD,OAAO,EACH,SAAS,EACT,eAAe,EACf,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,EAChB,cAAc,EACd,mBAAmB,EACnB,yBAAyB,EACzB,0BAA0B,EAC1B,mBAAmB,EACnB,mBAAmB,EACnB,sBAAsB,EACtB,gBAAgB,GACnB,MAAM,0BAA0B,CAAC;AAElC,yCAAyC;AACzC,OAAO,EACH,KAAK,EACL,+BAA+B,EAC/B,eAAe,EACf,sBAAsB,EACtB,gCAAgC,EAChC,iBAAiB,EACjB,UAAU,EACV,mBAAmB,EACnB,aAAa,EACb,eAAe,EACf,kBAAkB,EAClB,wBAAwB,EACxB,kBAAkB,EAClB,yBAAyB,EACzB,sBAAsB,EACtB,uBAAuB,EACvB,kBAAkB,EAClB,uBAAuB,GAC1B,MAAM,kBAAkB,CAAC;AAE1B,4CAA4C;AAC5C,OAAO,EACH,IAAI,EACJ,aAAa,EACb,SAAS,EACT,4BAA4B,EAC5B,wBAAwB,EACxB,uBAAuB,EACvB,uBAAuB,EACvB,qBAAqB,EACrB,iBAAiB,EACjB,cAAc,EACd,cAAc,GACjB,MAAM,gBAAgB,CAAC;AAExB,eAAe;AACf,OAAO,EACH,UAAU,IAAI,gBAAgB,EAC9B,UAAU,IAAI,gBAAgB,EAC9B,mBAAmB,IAAI,yBAAyB,EAChD,mBAAmB,EACnB,eAAe,GAClB,MAAM,YAAY,CAAC"}
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,aAAa,EACb,2BAA2B,EAC3B,iCAAiC,EACjC,0BAA0B,EAC1B,gCAAgC,EAChC,uBAAuB,EACvB,6BAA6B,EAC7B,sCAAsC,EACtC,4CAA4C,EAC5C,6BAA6B,EAC7B,+BAA+B,EAC/B,YAAY,EACZ,eAAe,EACf,eAAe,EACf,oBAAoB,EACpB,4BAA4B,EAC5B,wBAAwB;AACxB,yBAAyB;AACzB,oBAAoB,EACpB,gBAAgB,EAChB,iBAAiB,EACjB,qBAAqB,EACrB,sBAAsB,EACtB,qBAAqB,EACrB,sBAAsB,EACtB,wBAAwB,EACxB,yBAAyB,EACzB,+BAA+B,EAC/B,8BAA8B,EAC9B,+BAA+B,EAC/B,6BAA6B,EAC7B,mCAAmC,EACnC,oCAAoC,GACvC,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,sBAAsB,EACtB,eAAe,EACf,qBAAqB,EACrB,qBAAqB,EACrB,wBAAwB,EACxB,YAAY,EACZ,iBAAiB,EACjB,yBAAyB,EACzB,gCAAgC,EAChC,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,gDAAgD;AAChD,OAAO,EACH,SAAS,EACT,eAAe,EACf,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,EAChB,cAAc,EACd,mBAAmB,EACnB,yBAAyB,EACzB,0BAA0B,EAC1B,mBAAmB,EACnB,mBAAmB,EACnB,sBAAsB,EACtB,gBAAgB,GACnB,MAAM,0BAA0B,CAAC;AAElC,yCAAyC;AACzC,OAAO,EACH,KAAK,EACL,+BAA+B,EAC/B,eAAe,EACf,sBAAsB,EACtB,gCAAgC,EAChC,iBAAiB,EACjB,UAAU,EACV,mBAAmB,EACnB,aAAa,EACb,eAAe,EACf,kBAAkB,EAClB,wBAAwB,EACxB,kBAAkB,EAClB,yBAAyB,EACzB,sBAAsB,EACtB,uBAAuB,EACvB,kBAAkB,EAClB,uBAAuB,GAC1B,MAAM,kBAAkB,CAAC;AAE1B,4CAA4C;AAC5C,OAAO,EACH,IAAI,EACJ,aAAa,EACb,SAAS,EACT,4BAA4B,EAC5B,wBAAwB,EACxB,uBAAuB,EACvB,uBAAuB,EACvB,qBAAqB,EACrB,iBAAiB,EACjB,cAAc,EACd,cAAc,GACjB,MAAM,gBAAgB,CAAC;AAExB,eAAe;AACf,OAAO,EACH,UAAU,IAAI,gBAAgB,EAC9B,UAAU,IAAI,gBAAgB,EAC9B,mBAAmB,IAAI,yBAAyB,EAChD,mBAAmB,EACnB,eAAe,GAClB,MAAM,YAAY,CAAC"}
package/dist/sklearn/sklearn.d.ts CHANGED
@@ -33,69 +33,44 @@ export { LightGBMConfigType } from "../lightgbm/lightgbm.js";
33
33
  export { NGBoostConfigType } from "../ngboost/ngboost.js";
34
34
  export { GPConfigType } from "../gp/gp.js";
35
35
  /**
36
- * Configuration for train/test split.
36
+ * Configuration for data splitting.
37
+ *
38
+ * Examples:
39
+ * - 2-way: split_sizes: [0.8, 0.2] -> train/test
40
+ * - 3-way: split_sizes: [0.7, 0.15, 0.15] -> train/val/test
41
+ * - 4-way: split_sizes: [0.6, 0.1, 0.15, 0.15] -> train/val/calib/test
37
42
  */
38
43
  export declare const SplitConfigType: StructType<{
39
- /** Proportion of data to include in test split (default 0.2) */
40
- readonly test_size: OptionType<FloatType>;
41
- /** Random seed for reproducibility */
42
- readonly random_state: OptionType<IntegerType>;
43
- /** Whether to shuffle data before splitting (default true) */
44
- readonly shuffle: OptionType<BooleanType>;
45
- /** Stratification labels (same length as X). Ensures proportional representation in each split. */
46
- readonly stratify: OptionType<ArrayType<IntegerType>>;
47
- /** Minimum samples per stratify class. Classes with fewer samples are rejected. (default 2) */
48
- readonly min_stratify_samples: OptionType<IntegerType>;
49
- }>;
50
- /**
51
- * Configuration for 3-way train/val/test split.
52
- */
53
- export declare const ThreeWaySplitConfigType: StructType<{
54
- /** Proportion of data for validation (default 0.15) */
55
- readonly val_size: OptionType<FloatType>;
56
- /** Proportion of data for test/holdout (default 0.15) */
57
- readonly test_size: OptionType<FloatType>;
44
+ /** Array of split proportions (must sum to 1.0). */
45
+ readonly split_sizes: ArrayType<FloatType>;
58
46
  /** Random seed for reproducibility */
59
47
  readonly random_state: OptionType<IntegerType>;
60
48
  /** Whether to shuffle data before splitting (default true) */
61
49
  readonly shuffle: OptionType<BooleanType>;
62
- /** Stratification labels (same length as X). Ensures proportional representation in each split. */
63
- readonly stratify: OptionType<ArrayType<IntegerType>>;
64
- /** Minimum samples per stratify class. Classes with fewer samples are rejected. (default 3) */
50
+ /**
51
+ * Multiple stratification columns - combined into compound strata.
52
+ * Each inner array is one column of labels (same length as X).
53
+ * E.g., [[origin1, origin2, ...], [mpf1, mpf2, ...]] stratifies on origin × mpf.
54
+ */
55
+ readonly stratify: OptionType<ArrayType<ArrayType<IntegerType>>>;
56
+ /** Minimum samples per stratify class. Classes with fewer samples are rejected. (default = n_splits) */
65
57
  readonly min_stratify_samples: OptionType<IntegerType>;
58
+ /**
59
+ * Columns that must have overlapping representation in all splits (but not used for stratification).
60
+ * Samples with values that don't appear in all splits are rejected.
61
+ * Each inner array is one column of labels (same length as X).
62
+ */
63
+ readonly overlap: OptionType<ArrayType<ArrayType<IntegerType>>>;
66
64
  }>;
67
65
  /**
68
- * Result of train/test split.
66
+ * Result of data splitting.
69
67
  */
70
68
  export declare const SplitResultType: StructType<{
71
- /** Training features */
72
- readonly X_train: ArrayType<ArrayType<FloatType>>;
73
- /** Test features */
74
- readonly X_test: ArrayType<ArrayType<FloatType>>;
75
- /** Training labels */
76
- readonly y_train: ArrayType<FloatType>;
77
- /** Test labels */
78
- readonly y_test: ArrayType<FloatType>;
79
- /** Indices of rows rejected due to rare stratify classes (empty if no stratify or no rejections) */
80
- readonly rejected_indices: ArrayType<IntegerType>;
81
- }>;
82
- /**
83
- * Result of 3-way train/val/test split.
84
- */
85
- export declare const ThreeWaySplitResultType: StructType<{
86
- /** Training features */
87
- readonly X_train: ArrayType<ArrayType<FloatType>>;
88
- /** Validation features */
89
- readonly X_val: ArrayType<ArrayType<FloatType>>;
90
- /** Test/holdout features */
91
- readonly X_test: ArrayType<ArrayType<FloatType>>;
92
- /** Training targets (matrix) */
93
- readonly Y_train: ArrayType<ArrayType<FloatType>>;
94
- /** Validation targets (matrix) */
95
- readonly Y_val: ArrayType<ArrayType<FloatType>>;
96
- /** Test/holdout targets (matrix) */
97
- readonly Y_test: ArrayType<ArrayType<FloatType>>;
98
- /** Indices of rows rejected due to rare stratify classes (empty if no stratify or no rejections) */
69
+ /** Array of feature matrices, one per split (in order of split_sizes) */
70
+ readonly X_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
71
+ /** Array of target matrices, one per split (in order of split_sizes) */
72
+ readonly Y_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
73
+ /** Indices of rows rejected due to rare stratify classes or missing overlap values */
99
74
  readonly rejected_indices: ArrayType<IntegerType>;
100
75
  }>;
101
76
  /**
@@ -749,34 +724,57 @@ export declare const RegressorChainConfigType: StructType<{
749
724
  readonly random_state: OptionType<IntegerType>;
750
725
  }>;
751
726
  /**
752
- * Split arrays into train and test subsets.
727
+ * Split arrays into N subsets (train/test, train/val/test, etc.).
753
728
  *
754
729
  * @param X - Feature matrix
755
- * @param y - Target vector
756
- * @param config - Split configuration
757
- * @returns Split result with X_train, X_test, y_train, y_test
730
+ * @param Y - Target matrix
731
+ * @param config - Split configuration with split_sizes, stratify, overlap
732
+ * @returns Split result with X_splits, Y_splits arrays
733
+ *
734
+ * @example
735
+ * ```ts
736
+ * // 2-way split (train/test)
737
+ * const result = Sklearn.split(X, Y, { split_sizes: [0.8, 0.2], ... });
738
+ * const [X_train, X_test] = [result.X_splits.get(0n), result.X_splits.get(1n)];
739
+ *
740
+ * // 3-way split (train/val/test)
741
+ * const result = Sklearn.split(X, Y, { split_sizes: [0.7, 0.15, 0.15], ... });
742
+ *
743
+ * // With multi-column stratification
744
+ * const result = Sklearn.split(X, Y, {
745
+ * split_sizes: [0.7, 0.15, 0.15],
746
+ * stratify: variant('some', [origin_labels, mpf_labels]),
747
+ * overlap: variant('some', [class_labels]),
748
+ * });
749
+ * ```
758
750
  */
759
- export declare const sklearn_train_test_split: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<FloatType>, StructType<{
760
- /** Proportion of data to include in test split (default 0.2) */
761
- readonly test_size: OptionType<FloatType>;
751
+ export declare const sklearn_split: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, StructType<{
752
+ /** Array of split proportions (must sum to 1.0). */
753
+ readonly split_sizes: ArrayType<FloatType>;
762
754
  /** Random seed for reproducibility */
763
755
  readonly random_state: OptionType<IntegerType>;
764
756
  /** Whether to shuffle data before splitting (default true) */
765
757
  readonly shuffle: OptionType<BooleanType>;
766
- /** Stratification labels (same length as X). Ensures proportional representation in each split. */
767
- readonly stratify: OptionType<ArrayType<IntegerType>>;
768
- /** Minimum samples per stratify class. Classes with fewer samples are rejected. (default 2) */
758
+ /**
759
+ * Multiple stratification columns - combined into compound strata.
760
+ * Each inner array is one column of labels (same length as X).
761
+ * E.g., [[origin1, origin2, ...], [mpf1, mpf2, ...]] stratifies on origin × mpf.
762
+ */
763
+ readonly stratify: OptionType<ArrayType<ArrayType<IntegerType>>>;
764
+ /** Minimum samples per stratify class. Classes with fewer samples are rejected. (default = n_splits) */
769
765
  readonly min_stratify_samples: OptionType<IntegerType>;
766
+ /**
767
+ * Columns that must have overlapping representation in all splits (but not used for stratification).
768
+ * Samples with values that don't appear in all splits are rejected.
769
+ * Each inner array is one column of labels (same length as X).
770
+ */
771
+ readonly overlap: OptionType<ArrayType<ArrayType<IntegerType>>>;
770
772
  }>], StructType<{
771
- /** Training features */
772
- readonly X_train: ArrayType<ArrayType<FloatType>>;
773
- /** Test features */
774
- readonly X_test: ArrayType<ArrayType<FloatType>>;
775
- /** Training labels */
776
- readonly y_train: ArrayType<FloatType>;
777
- /** Test labels */
778
- readonly y_test: ArrayType<FloatType>;
779
- /** Indices of rows rejected due to rare stratify classes (empty if no stratify or no rejections) */
773
+ /** Array of feature matrices, one per split (in order of split_sizes) */
774
+ readonly X_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
775
+ /** Array of target matrices, one per split (in order of split_sizes) */
776
+ readonly Y_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
777
+ /** Indices of rows rejected due to rare stratify classes or missing overlap values */
780
778
  readonly rejected_indices: ArrayType<IntegerType>;
781
779
  }>>;
782
780
  /**
@@ -1640,43 +1638,6 @@ export declare const sklearn_regressor_chain_predict: import("@elaraai/east").Pl
1640
1638
  readonly base_estimator_type: StringType;
1641
1639
  }>;
1642
1640
  }>, ArrayType<ArrayType<FloatType>>], ArrayType<ArrayType<FloatType>>>;
1643
- /**
1644
- * Split arrays into train, validation, and test subsets.
1645
- *
1646
- * @param X - Feature matrix
1647
- * @param Y - Target matrix (multi-target)
1648
- * @param config - Split configuration with val_size and test_size
1649
- * @returns Split result with X_train, X_val, X_test, Y_train, Y_val, Y_test
1650
- */
1651
- export declare const sklearn_train_val_test_split: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, StructType<{
1652
- /** Proportion of data for validation (default 0.15) */
1653
- readonly val_size: OptionType<FloatType>;
1654
- /** Proportion of data for test/holdout (default 0.15) */
1655
- readonly test_size: OptionType<FloatType>;
1656
- /** Random seed for reproducibility */
1657
- readonly random_state: OptionType<IntegerType>;
1658
- /** Whether to shuffle data before splitting (default true) */
1659
- readonly shuffle: OptionType<BooleanType>;
1660
- /** Stratification labels (same length as X). Ensures proportional representation in each split. */
1661
- readonly stratify: OptionType<ArrayType<IntegerType>>;
1662
- /** Minimum samples per stratify class. Classes with fewer samples are rejected. (default 3) */
1663
- readonly min_stratify_samples: OptionType<IntegerType>;
1664
- }>], StructType<{
1665
- /** Training features */
1666
- readonly X_train: ArrayType<ArrayType<FloatType>>;
1667
- /** Validation features */
1668
- readonly X_val: ArrayType<ArrayType<FloatType>>;
1669
- /** Test/holdout features */
1670
- readonly X_test: ArrayType<ArrayType<FloatType>>;
1671
- /** Training targets (matrix) */
1672
- readonly Y_train: ArrayType<ArrayType<FloatType>>;
1673
- /** Validation targets (matrix) */
1674
- readonly Y_val: ArrayType<ArrayType<FloatType>>;
1675
- /** Test/holdout targets (matrix) */
1676
- readonly Y_test: ArrayType<ArrayType<FloatType>>;
1677
- /** Indices of rows rejected due to rare stratify classes (empty if no stratify or no rejections) */
1678
- readonly rejected_indices: ArrayType<IntegerType>;
1679
- }>>;
1680
1641
  /**
1681
1642
  * Compute regression metrics for single-target predictions.
1682
1643
  *
@@ -2033,60 +1994,34 @@ export declare const SklearnTypes: {
2033
1994
  }>;
2034
1995
  /** Split configuration type */
2035
1996
  readonly SplitConfigType: StructType<{
2036
- /** Proportion of data to include in test split (default 0.2) */
2037
- readonly test_size: OptionType<FloatType>;
1997
+ /** Array of split proportions (must sum to 1.0). */
1998
+ readonly split_sizes: ArrayType<FloatType>;
2038
1999
  /** Random seed for reproducibility */
2039
2000
  readonly random_state: OptionType<IntegerType>;
2040
2001
  /** Whether to shuffle data before splitting (default true) */
2041
2002
  readonly shuffle: OptionType<BooleanType>;
2042
- /** Stratification labels (same length as X). Ensures proportional representation in each split. */
2043
- readonly stratify: OptionType<ArrayType<IntegerType>>;
2044
- /** Minimum samples per stratify class. Classes with fewer samples are rejected. (default 2) */
2003
+ /**
2004
+ * Multiple stratification columns - combined into compound strata.
2005
+ * Each inner array is one column of labels (same length as X).
2006
+ * E.g., [[origin1, origin2, ...], [mpf1, mpf2, ...]] stratifies on origin × mpf.
2007
+ */
2008
+ readonly stratify: OptionType<ArrayType<ArrayType<IntegerType>>>;
2009
+ /** Minimum samples per stratify class. Classes with fewer samples are rejected. (default = n_splits) */
2045
2010
  readonly min_stratify_samples: OptionType<IntegerType>;
2011
+ /**
2012
+ * Columns that must have overlapping representation in all splits (but not used for stratification).
2013
+ * Samples with values that don't appear in all splits are rejected.
2014
+ * Each inner array is one column of labels (same length as X).
2015
+ */
2016
+ readonly overlap: OptionType<ArrayType<ArrayType<IntegerType>>>;
2046
2017
  }>;
2047
2018
  /** Split result type */
2048
2019
  readonly SplitResultType: StructType<{
2049
- /** Training features */
2050
- readonly X_train: ArrayType<ArrayType<FloatType>>;
2051
- /** Test features */
2052
- readonly X_test: ArrayType<ArrayType<FloatType>>;
2053
- /** Training labels */
2054
- readonly y_train: ArrayType<FloatType>;
2055
- /** Test labels */
2056
- readonly y_test: ArrayType<FloatType>;
2057
- /** Indices of rows rejected due to rare stratify classes (empty if no stratify or no rejections) */
2058
- readonly rejected_indices: ArrayType<IntegerType>;
2059
- }>;
2060
- /** 3-way split configuration type */
2061
- readonly ThreeWaySplitConfigType: StructType<{
2062
- /** Proportion of data for validation (default 0.15) */
2063
- readonly val_size: OptionType<FloatType>;
2064
- /** Proportion of data for test/holdout (default 0.15) */
2065
- readonly test_size: OptionType<FloatType>;
2066
- /** Random seed for reproducibility */
2067
- readonly random_state: OptionType<IntegerType>;
2068
- /** Whether to shuffle data before splitting (default true) */
2069
- readonly shuffle: OptionType<BooleanType>;
2070
- /** Stratification labels (same length as X). Ensures proportional representation in each split. */
2071
- readonly stratify: OptionType<ArrayType<IntegerType>>;
2072
- /** Minimum samples per stratify class. Classes with fewer samples are rejected. (default 3) */
2073
- readonly min_stratify_samples: OptionType<IntegerType>;
2074
- }>;
2075
- /** 3-way split result type */
2076
- readonly ThreeWaySplitResultType: StructType<{
2077
- /** Training features */
2078
- readonly X_train: ArrayType<ArrayType<FloatType>>;
2079
- /** Validation features */
2080
- readonly X_val: ArrayType<ArrayType<FloatType>>;
2081
- /** Test/holdout features */
2082
- readonly X_test: ArrayType<ArrayType<FloatType>>;
2083
- /** Training targets (matrix) */
2084
- readonly Y_train: ArrayType<ArrayType<FloatType>>;
2085
- /** Validation targets (matrix) */
2086
- readonly Y_val: ArrayType<ArrayType<FloatType>>;
2087
- /** Test/holdout targets (matrix) */
2088
- readonly Y_test: ArrayType<ArrayType<FloatType>>;
2089
- /** Indices of rows rejected due to rare stratify classes (empty if no stratify or no rejections) */
2020
+ /** Array of feature matrices, one per split (in order of split_sizes) */
2021
+ readonly X_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
2022
+ /** Array of target matrices, one per split (in order of split_sizes) */
2023
+ readonly Y_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
2024
+ /** Indices of rows rejected due to rare stratify classes or missing overlap values */
2090
2025
  readonly rejected_indices: ArrayType<IntegerType>;
2091
2026
  }>;
2092
2027
  /** Model blob type for sklearn models */
@@ -2690,58 +2625,34 @@ export declare const SklearnTypes: {
2690
2625
  * ```
2691
2626
  */
2692
2627
  export declare const Sklearn: {
2693
- /** Split arrays into train and test subsets */
2694
- readonly trainTestSplit: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<FloatType>, StructType<{
2695
- /** Proportion of data to include in test split (default 0.2) */
2696
- readonly test_size: OptionType<FloatType>;
2697
- /** Random seed for reproducibility */
2698
- readonly random_state: OptionType<IntegerType>;
2699
- /** Whether to shuffle data before splitting (default true) */
2700
- readonly shuffle: OptionType<BooleanType>;
2701
- /** Stratification labels (same length as X). Ensures proportional representation in each split. */
2702
- readonly stratify: OptionType<ArrayType<IntegerType>>;
2703
- /** Minimum samples per stratify class. Classes with fewer samples are rejected. (default 2) */
2704
- readonly min_stratify_samples: OptionType<IntegerType>;
2705
- }>], StructType<{
2706
- /** Training features */
2707
- readonly X_train: ArrayType<ArrayType<FloatType>>;
2708
- /** Test features */
2709
- readonly X_test: ArrayType<ArrayType<FloatType>>;
2710
- /** Training labels */
2711
- readonly y_train: ArrayType<FloatType>;
2712
- /** Test labels */
2713
- readonly y_test: ArrayType<FloatType>;
2714
- /** Indices of rows rejected due to rare stratify classes (empty if no stratify or no rejections) */
2715
- readonly rejected_indices: ArrayType<IntegerType>;
2716
- }>>;
2717
- /** Split arrays into train, validation, and test subsets */
2718
- readonly trainValTestSplit: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, StructType<{
2719
- /** Proportion of data for validation (default 0.15) */
2720
- readonly val_size: OptionType<FloatType>;
2721
- /** Proportion of data for test/holdout (default 0.15) */
2722
- readonly test_size: OptionType<FloatType>;
2628
+ /** Split arrays into N subsets (train/test, train/val/test, etc.) */
2629
+ readonly split: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, StructType<{
2630
+ /** Array of split proportions (must sum to 1.0). */
2631
+ readonly split_sizes: ArrayType<FloatType>;
2723
2632
  /** Random seed for reproducibility */
2724
2633
  readonly random_state: OptionType<IntegerType>;
2725
2634
  /** Whether to shuffle data before splitting (default true) */
2726
2635
  readonly shuffle: OptionType<BooleanType>;
2727
- /** Stratification labels (same length as X). Ensures proportional representation in each split. */
2728
- readonly stratify: OptionType<ArrayType<IntegerType>>;
2729
- /** Minimum samples per stratify class. Classes with fewer samples are rejected. (default 3) */
2636
+ /**
2637
+ * Multiple stratification columns - combined into compound strata.
2638
+ * Each inner array is one column of labels (same length as X).
2639
+ * E.g., [[origin1, origin2, ...], [mpf1, mpf2, ...]] stratifies on origin × mpf.
2640
+ */
2641
+ readonly stratify: OptionType<ArrayType<ArrayType<IntegerType>>>;
2642
+ /** Minimum samples per stratify class. Classes with fewer samples are rejected. (default = n_splits) */
2730
2643
  readonly min_stratify_samples: OptionType<IntegerType>;
2644
+ /**
2645
+ * Columns that must have overlapping representation in all splits (but not used for stratification).
2646
+ * Samples with values that don't appear in all splits are rejected.
2647
+ * Each inner array is one column of labels (same length as X).
2648
+ */
2649
+ readonly overlap: OptionType<ArrayType<ArrayType<IntegerType>>>;
2731
2650
  }>], StructType<{
2732
- /** Training features */
2733
- readonly X_train: ArrayType<ArrayType<FloatType>>;
2734
- /** Validation features */
2735
- readonly X_val: ArrayType<ArrayType<FloatType>>;
2736
- /** Test/holdout features */
2737
- readonly X_test: ArrayType<ArrayType<FloatType>>;
2738
- /** Training targets (matrix) */
2739
- readonly Y_train: ArrayType<ArrayType<FloatType>>;
2740
- /** Validation targets (matrix) */
2741
- readonly Y_val: ArrayType<ArrayType<FloatType>>;
2742
- /** Test/holdout targets (matrix) */
2743
- readonly Y_test: ArrayType<ArrayType<FloatType>>;
2744
- /** Indices of rows rejected due to rare stratify classes (empty if no stratify or no rejections) */
2651
+ /** Array of feature matrices, one per split (in order of split_sizes) */
2652
+ readonly X_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
2653
+ /** Array of target matrices, one per split (in order of split_sizes) */
2654
+ readonly Y_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
2655
+ /** Indices of rows rejected due to rare stratify classes or missing overlap values */
2745
2656
  readonly rejected_indices: ArrayType<IntegerType>;
2746
2657
  }>>;
2747
2658
  /** Fit a StandardScaler to data */
@@ -3806,60 +3717,34 @@ export declare const Sklearn: {
3806
3717
  }>;
3807
3718
  /** Split configuration type */
3808
3719
  readonly SplitConfigType: StructType<{
3809
- /** Proportion of data to include in test split (default 0.2) */
3810
- readonly test_size: OptionType<FloatType>;
3720
+ /** Array of split proportions (must sum to 1.0). */
3721
+ readonly split_sizes: ArrayType<FloatType>;
3811
3722
  /** Random seed for reproducibility */
3812
3723
  readonly random_state: OptionType<IntegerType>;
3813
3724
  /** Whether to shuffle data before splitting (default true) */
3814
3725
  readonly shuffle: OptionType<BooleanType>;
3815
- /** Stratification labels (same length as X). Ensures proportional representation in each split. */
3816
- readonly stratify: OptionType<ArrayType<IntegerType>>;
3817
- /** Minimum samples per stratify class. Classes with fewer samples are rejected. (default 2) */
3726
+ /**
3727
+ * Multiple stratification columns - combined into compound strata.
3728
+ * Each inner array is one column of labels (same length as X).
3729
+ * E.g., [[origin1, origin2, ...], [mpf1, mpf2, ...]] stratifies on origin × mpf.
3730
+ */
3731
+ readonly stratify: OptionType<ArrayType<ArrayType<IntegerType>>>;
3732
+ /** Minimum samples per stratify class. Classes with fewer samples are rejected. (default = n_splits) */
3818
3733
  readonly min_stratify_samples: OptionType<IntegerType>;
3734
+ /**
3735
+ * Columns that must have overlapping representation in all splits (but not used for stratification).
3736
+ * Samples with values that don't appear in all splits are rejected.
3737
+ * Each inner array is one column of labels (same length as X).
3738
+ */
3739
+ readonly overlap: OptionType<ArrayType<ArrayType<IntegerType>>>;
3819
3740
  }>;
3820
3741
  /** Split result type */
3821
3742
  readonly SplitResultType: StructType<{
3822
- /** Training features */
3823
- readonly X_train: ArrayType<ArrayType<FloatType>>;
3824
- /** Test features */
3825
- readonly X_test: ArrayType<ArrayType<FloatType>>;
3826
- /** Training labels */
3827
- readonly y_train: ArrayType<FloatType>;
3828
- /** Test labels */
3829
- readonly y_test: ArrayType<FloatType>;
3830
- /** Indices of rows rejected due to rare stratify classes (empty if no stratify or no rejections) */
3831
- readonly rejected_indices: ArrayType<IntegerType>;
3832
- }>;
3833
- /** 3-way split configuration type */
3834
- readonly ThreeWaySplitConfigType: StructType<{
3835
- /** Proportion of data for validation (default 0.15) */
3836
- readonly val_size: OptionType<FloatType>;
3837
- /** Proportion of data for test/holdout (default 0.15) */
3838
- readonly test_size: OptionType<FloatType>;
3839
- /** Random seed for reproducibility */
3840
- readonly random_state: OptionType<IntegerType>;
3841
- /** Whether to shuffle data before splitting (default true) */
3842
- readonly shuffle: OptionType<BooleanType>;
3843
- /** Stratification labels (same length as X). Ensures proportional representation in each split. */
3844
- readonly stratify: OptionType<ArrayType<IntegerType>>;
3845
- /** Minimum samples per stratify class. Classes with fewer samples are rejected. (default 3) */
3846
- readonly min_stratify_samples: OptionType<IntegerType>;
3847
- }>;
3848
- /** 3-way split result type */
3849
- readonly ThreeWaySplitResultType: StructType<{
3850
- /** Training features */
3851
- readonly X_train: ArrayType<ArrayType<FloatType>>;
3852
- /** Validation features */
3853
- readonly X_val: ArrayType<ArrayType<FloatType>>;
3854
- /** Test/holdout features */
3855
- readonly X_test: ArrayType<ArrayType<FloatType>>;
3856
- /** Training targets (matrix) */
3857
- readonly Y_train: ArrayType<ArrayType<FloatType>>;
3858
- /** Validation targets (matrix) */
3859
- readonly Y_val: ArrayType<ArrayType<FloatType>>;
3860
- /** Test/holdout targets (matrix) */
3861
- readonly Y_test: ArrayType<ArrayType<FloatType>>;
3862
- /** Indices of rows rejected due to rare stratify classes (empty if no stratify or no rejections) */
3743
+ /** Array of feature matrices, one per split (in order of split_sizes) */
3744
+ readonly X_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
3745
+ /** Array of target matrices, one per split (in order of split_sizes) */
3746
+ readonly Y_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
3747
+ /** Indices of rows rejected due to rare stratify classes or missing overlap values */
3863
3748
  readonly rejected_indices: ArrayType<IntegerType>;
3864
3749
  }>;
3865
3750
  /** Model blob type for sklearn models */