@elaraai/east-py-datascience 0.0.2-beta.55 → 0.0.2-beta.57
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.
- package/dist/index.d.ts +1 -1
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +1 -1
- package/dist/index.js.map +1 -1
- package/dist/sklearn/sklearn.d.ts +151 -251
- package/dist/sklearn/sklearn.d.ts.map +1 -1
- package/dist/sklearn/sklearn.js +55 -75
- package/dist/sklearn/sklearn.js.map +1 -1
- package/dist/tsconfig.tsbuildinfo +1 -1
- 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,
|
|
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,
|
|
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,
|
|
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,
|
|
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"}
|
|
@@ -33,69 +33,47 @@ 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
|
|
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
|
-
/**
|
|
40
|
-
readonly
|
|
44
|
+
/** Array of split proportions (must sum to 1.0). */
|
|
45
|
+
readonly split_sizes: ArrayType<FloatType>;
|
|
41
46
|
/** Random seed for reproducibility */
|
|
42
47
|
readonly random_state: OptionType<IntegerType>;
|
|
43
48
|
/** Whether to shuffle data before splitting (default true) */
|
|
44
49
|
readonly shuffle: OptionType<BooleanType>;
|
|
45
|
-
/**
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
50
|
+
/**
|
|
51
|
+
* Stratification columns - controls proportional distribution across splits.
|
|
52
|
+
* Combined into compound strata. Each inner array is one column of labels.
|
|
53
|
+
* Note: Stratify does NOT guarantee overlap - use the overlap parameter for that.
|
|
54
|
+
*/
|
|
55
|
+
readonly stratify: OptionType<ArrayType<ArrayType<IntegerType>>>;
|
|
56
|
+
/**
|
|
57
|
+
* Columns that must have overlapping representation in all splits.
|
|
58
|
+
* Each column is checked independently - values that don't appear in all splits are rejected.
|
|
59
|
+
* Each inner array is one column of labels (same length as X).
|
|
60
|
+
*/
|
|
61
|
+
readonly overlap: OptionType<ArrayType<ArrayType<IntegerType>>>;
|
|
62
|
+
/**
|
|
63
|
+
* Minimum samples per overlap value. Values with fewer samples are rejected. (default = n_splits)
|
|
64
|
+
* This ensures enough samples to distribute across all splits.
|
|
65
|
+
*/
|
|
66
|
+
readonly min_overlap: OptionType<IntegerType>;
|
|
49
67
|
}>;
|
|
50
68
|
/**
|
|
51
|
-
*
|
|
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>;
|
|
58
|
-
/** Random seed for reproducibility */
|
|
59
|
-
readonly random_state: OptionType<IntegerType>;
|
|
60
|
-
/** Whether to shuffle data before splitting (default true) */
|
|
61
|
-
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) */
|
|
65
|
-
readonly min_stratify_samples: OptionType<IntegerType>;
|
|
66
|
-
}>;
|
|
67
|
-
/**
|
|
68
|
-
* Result of train/test split.
|
|
69
|
+
* Result of data splitting.
|
|
69
70
|
*/
|
|
70
71
|
export declare const SplitResultType: StructType<{
|
|
71
|
-
/**
|
|
72
|
-
readonly
|
|
73
|
-
/**
|
|
74
|
-
readonly
|
|
75
|
-
/**
|
|
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) */
|
|
72
|
+
/** Array of feature matrices, one per split (in order of split_sizes) */
|
|
73
|
+
readonly X_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
|
|
74
|
+
/** Array of target matrices, one per split (in order of split_sizes) */
|
|
75
|
+
readonly Y_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
|
|
76
|
+
/** Indices of rows rejected due to rare stratify classes or missing overlap values */
|
|
99
77
|
readonly rejected_indices: ArrayType<IntegerType>;
|
|
100
78
|
}>;
|
|
101
79
|
/**
|
|
@@ -749,34 +727,60 @@ export declare const RegressorChainConfigType: StructType<{
|
|
|
749
727
|
readonly random_state: OptionType<IntegerType>;
|
|
750
728
|
}>;
|
|
751
729
|
/**
|
|
752
|
-
* Split arrays into
|
|
730
|
+
* Split arrays into N subsets (train/test, train/val/test, etc.).
|
|
753
731
|
*
|
|
754
732
|
* @param X - Feature matrix
|
|
755
|
-
* @param
|
|
756
|
-
* @param config - Split configuration
|
|
757
|
-
* @returns Split result with
|
|
733
|
+
* @param Y - Target matrix
|
|
734
|
+
* @param config - Split configuration with split_sizes, stratify, overlap
|
|
735
|
+
* @returns Split result with X_splits, Y_splits arrays
|
|
736
|
+
*
|
|
737
|
+
* @example
|
|
738
|
+
* ```ts
|
|
739
|
+
* // 2-way split (train/test)
|
|
740
|
+
* const result = Sklearn.split(X, Y, { split_sizes: [0.8, 0.2], ... });
|
|
741
|
+
* const [X_train, X_test] = [result.X_splits.get(0n), result.X_splits.get(1n)];
|
|
742
|
+
*
|
|
743
|
+
* // 3-way split (train/val/test)
|
|
744
|
+
* const result = Sklearn.split(X, Y, { split_sizes: [0.7, 0.15, 0.15], ... });
|
|
745
|
+
*
|
|
746
|
+
* // With multi-column stratification
|
|
747
|
+
* const result = Sklearn.split(X, Y, {
|
|
748
|
+
* split_sizes: [0.7, 0.15, 0.15],
|
|
749
|
+
* stratify: variant('some', [origin_labels, mpf_labels]),
|
|
750
|
+
* overlap: variant('some', [class_labels]),
|
|
751
|
+
* });
|
|
752
|
+
* ```
|
|
758
753
|
*/
|
|
759
|
-
export declare const
|
|
760
|
-
/**
|
|
761
|
-
readonly
|
|
754
|
+
export declare const sklearn_split: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, StructType<{
|
|
755
|
+
/** Array of split proportions (must sum to 1.0). */
|
|
756
|
+
readonly split_sizes: ArrayType<FloatType>;
|
|
762
757
|
/** Random seed for reproducibility */
|
|
763
758
|
readonly random_state: OptionType<IntegerType>;
|
|
764
759
|
/** Whether to shuffle data before splitting (default true) */
|
|
765
760
|
readonly shuffle: OptionType<BooleanType>;
|
|
766
|
-
/**
|
|
767
|
-
|
|
768
|
-
|
|
769
|
-
|
|
761
|
+
/**
|
|
762
|
+
* Stratification columns - controls proportional distribution across splits.
|
|
763
|
+
* Combined into compound strata. Each inner array is one column of labels.
|
|
764
|
+
* Note: Stratify does NOT guarantee overlap - use the overlap parameter for that.
|
|
765
|
+
*/
|
|
766
|
+
readonly stratify: OptionType<ArrayType<ArrayType<IntegerType>>>;
|
|
767
|
+
/**
|
|
768
|
+
* Columns that must have overlapping representation in all splits.
|
|
769
|
+
* Each column is checked independently - values that don't appear in all splits are rejected.
|
|
770
|
+
* Each inner array is one column of labels (same length as X).
|
|
771
|
+
*/
|
|
772
|
+
readonly overlap: OptionType<ArrayType<ArrayType<IntegerType>>>;
|
|
773
|
+
/**
|
|
774
|
+
* Minimum samples per overlap value. Values with fewer samples are rejected. (default = n_splits)
|
|
775
|
+
* This ensures enough samples to distribute across all splits.
|
|
776
|
+
*/
|
|
777
|
+
readonly min_overlap: OptionType<IntegerType>;
|
|
770
778
|
}>], StructType<{
|
|
771
|
-
/**
|
|
772
|
-
readonly
|
|
773
|
-
/**
|
|
774
|
-
readonly
|
|
775
|
-
/**
|
|
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) */
|
|
779
|
+
/** Array of feature matrices, one per split (in order of split_sizes) */
|
|
780
|
+
readonly X_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
|
|
781
|
+
/** Array of target matrices, one per split (in order of split_sizes) */
|
|
782
|
+
readonly Y_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
|
|
783
|
+
/** Indices of rows rejected due to rare stratify classes or missing overlap values */
|
|
780
784
|
readonly rejected_indices: ArrayType<IntegerType>;
|
|
781
785
|
}>>;
|
|
782
786
|
/**
|
|
@@ -1640,43 +1644,6 @@ export declare const sklearn_regressor_chain_predict: import("@elaraai/east").Pl
|
|
|
1640
1644
|
readonly base_estimator_type: StringType;
|
|
1641
1645
|
}>;
|
|
1642
1646
|
}>, 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
1647
|
/**
|
|
1681
1648
|
* Compute regression metrics for single-target predictions.
|
|
1682
1649
|
*
|
|
@@ -2033,60 +2000,37 @@ export declare const SklearnTypes: {
|
|
|
2033
2000
|
}>;
|
|
2034
2001
|
/** Split configuration type */
|
|
2035
2002
|
readonly SplitConfigType: StructType<{
|
|
2036
|
-
/**
|
|
2037
|
-
readonly
|
|
2003
|
+
/** Array of split proportions (must sum to 1.0). */
|
|
2004
|
+
readonly split_sizes: ArrayType<FloatType>;
|
|
2038
2005
|
/** Random seed for reproducibility */
|
|
2039
2006
|
readonly random_state: OptionType<IntegerType>;
|
|
2040
2007
|
/** Whether to shuffle data before splitting (default true) */
|
|
2041
2008
|
readonly shuffle: OptionType<BooleanType>;
|
|
2042
|
-
/**
|
|
2043
|
-
|
|
2044
|
-
|
|
2045
|
-
|
|
2009
|
+
/**
|
|
2010
|
+
* Stratification columns - controls proportional distribution across splits.
|
|
2011
|
+
* Combined into compound strata. Each inner array is one column of labels.
|
|
2012
|
+
* Note: Stratify does NOT guarantee overlap - use the overlap parameter for that.
|
|
2013
|
+
*/
|
|
2014
|
+
readonly stratify: OptionType<ArrayType<ArrayType<IntegerType>>>;
|
|
2015
|
+
/**
|
|
2016
|
+
* Columns that must have overlapping representation in all splits.
|
|
2017
|
+
* Each column is checked independently - values that don't appear in all splits are rejected.
|
|
2018
|
+
* Each inner array is one column of labels (same length as X).
|
|
2019
|
+
*/
|
|
2020
|
+
readonly overlap: OptionType<ArrayType<ArrayType<IntegerType>>>;
|
|
2021
|
+
/**
|
|
2022
|
+
* Minimum samples per overlap value. Values with fewer samples are rejected. (default = n_splits)
|
|
2023
|
+
* This ensures enough samples to distribute across all splits.
|
|
2024
|
+
*/
|
|
2025
|
+
readonly min_overlap: OptionType<IntegerType>;
|
|
2046
2026
|
}>;
|
|
2047
2027
|
/** Split result type */
|
|
2048
2028
|
readonly SplitResultType: StructType<{
|
|
2049
|
-
/**
|
|
2050
|
-
readonly
|
|
2051
|
-
/**
|
|
2052
|
-
readonly
|
|
2053
|
-
/**
|
|
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) */
|
|
2029
|
+
/** Array of feature matrices, one per split (in order of split_sizes) */
|
|
2030
|
+
readonly X_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
|
|
2031
|
+
/** Array of target matrices, one per split (in order of split_sizes) */
|
|
2032
|
+
readonly Y_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
|
|
2033
|
+
/** Indices of rows rejected due to rare stratify classes or missing overlap values */
|
|
2090
2034
|
readonly rejected_indices: ArrayType<IntegerType>;
|
|
2091
2035
|
}>;
|
|
2092
2036
|
/** Model blob type for sklearn models */
|
|
@@ -2690,58 +2634,37 @@ export declare const SklearnTypes: {
|
|
|
2690
2634
|
* ```
|
|
2691
2635
|
*/
|
|
2692
2636
|
export declare const Sklearn: {
|
|
2693
|
-
/** Split arrays into
|
|
2694
|
-
readonly
|
|
2695
|
-
/**
|
|
2696
|
-
readonly
|
|
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>;
|
|
2637
|
+
/** Split arrays into N subsets (train/test, train/val/test, etc.) */
|
|
2638
|
+
readonly split: import("@elaraai/east").PlatformDefinition<[ArrayType<ArrayType<FloatType>>, ArrayType<ArrayType<FloatType>>, StructType<{
|
|
2639
|
+
/** Array of split proportions (must sum to 1.0). */
|
|
2640
|
+
readonly split_sizes: ArrayType<FloatType>;
|
|
2723
2641
|
/** Random seed for reproducibility */
|
|
2724
2642
|
readonly random_state: OptionType<IntegerType>;
|
|
2725
2643
|
/** Whether to shuffle data before splitting (default true) */
|
|
2726
2644
|
readonly shuffle: OptionType<BooleanType>;
|
|
2727
|
-
/**
|
|
2728
|
-
|
|
2729
|
-
|
|
2730
|
-
|
|
2645
|
+
/**
|
|
2646
|
+
* Stratification columns - controls proportional distribution across splits.
|
|
2647
|
+
* Combined into compound strata. Each inner array is one column of labels.
|
|
2648
|
+
* Note: Stratify does NOT guarantee overlap - use the overlap parameter for that.
|
|
2649
|
+
*/
|
|
2650
|
+
readonly stratify: OptionType<ArrayType<ArrayType<IntegerType>>>;
|
|
2651
|
+
/**
|
|
2652
|
+
* Columns that must have overlapping representation in all splits.
|
|
2653
|
+
* Each column is checked independently - values that don't appear in all splits are rejected.
|
|
2654
|
+
* Each inner array is one column of labels (same length as X).
|
|
2655
|
+
*/
|
|
2656
|
+
readonly overlap: OptionType<ArrayType<ArrayType<IntegerType>>>;
|
|
2657
|
+
/**
|
|
2658
|
+
* Minimum samples per overlap value. Values with fewer samples are rejected. (default = n_splits)
|
|
2659
|
+
* This ensures enough samples to distribute across all splits.
|
|
2660
|
+
*/
|
|
2661
|
+
readonly min_overlap: OptionType<IntegerType>;
|
|
2731
2662
|
}>], StructType<{
|
|
2732
|
-
/**
|
|
2733
|
-
readonly
|
|
2734
|
-
/**
|
|
2735
|
-
readonly
|
|
2736
|
-
/**
|
|
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) */
|
|
2663
|
+
/** Array of feature matrices, one per split (in order of split_sizes) */
|
|
2664
|
+
readonly X_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
|
|
2665
|
+
/** Array of target matrices, one per split (in order of split_sizes) */
|
|
2666
|
+
readonly Y_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
|
|
2667
|
+
/** Indices of rows rejected due to rare stratify classes or missing overlap values */
|
|
2745
2668
|
readonly rejected_indices: ArrayType<IntegerType>;
|
|
2746
2669
|
}>>;
|
|
2747
2670
|
/** Fit a StandardScaler to data */
|
|
@@ -3806,60 +3729,37 @@ export declare const Sklearn: {
|
|
|
3806
3729
|
}>;
|
|
3807
3730
|
/** Split configuration type */
|
|
3808
3731
|
readonly SplitConfigType: StructType<{
|
|
3809
|
-
/**
|
|
3810
|
-
readonly
|
|
3732
|
+
/** Array of split proportions (must sum to 1.0). */
|
|
3733
|
+
readonly split_sizes: ArrayType<FloatType>;
|
|
3811
3734
|
/** Random seed for reproducibility */
|
|
3812
3735
|
readonly random_state: OptionType<IntegerType>;
|
|
3813
3736
|
/** Whether to shuffle data before splitting (default true) */
|
|
3814
3737
|
readonly shuffle: OptionType<BooleanType>;
|
|
3815
|
-
/**
|
|
3816
|
-
|
|
3817
|
-
|
|
3818
|
-
|
|
3738
|
+
/**
|
|
3739
|
+
* Stratification columns - controls proportional distribution across splits.
|
|
3740
|
+
* Combined into compound strata. Each inner array is one column of labels.
|
|
3741
|
+
* Note: Stratify does NOT guarantee overlap - use the overlap parameter for that.
|
|
3742
|
+
*/
|
|
3743
|
+
readonly stratify: OptionType<ArrayType<ArrayType<IntegerType>>>;
|
|
3744
|
+
/**
|
|
3745
|
+
* Columns that must have overlapping representation in all splits.
|
|
3746
|
+
* Each column is checked independently - values that don't appear in all splits are rejected.
|
|
3747
|
+
* Each inner array is one column of labels (same length as X).
|
|
3748
|
+
*/
|
|
3749
|
+
readonly overlap: OptionType<ArrayType<ArrayType<IntegerType>>>;
|
|
3750
|
+
/**
|
|
3751
|
+
* Minimum samples per overlap value. Values with fewer samples are rejected. (default = n_splits)
|
|
3752
|
+
* This ensures enough samples to distribute across all splits.
|
|
3753
|
+
*/
|
|
3754
|
+
readonly min_overlap: OptionType<IntegerType>;
|
|
3819
3755
|
}>;
|
|
3820
3756
|
/** Split result type */
|
|
3821
3757
|
readonly SplitResultType: StructType<{
|
|
3822
|
-
/**
|
|
3823
|
-
readonly
|
|
3824
|
-
/**
|
|
3825
|
-
readonly
|
|
3826
|
-
/**
|
|
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) */
|
|
3758
|
+
/** Array of feature matrices, one per split (in order of split_sizes) */
|
|
3759
|
+
readonly X_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
|
|
3760
|
+
/** Array of target matrices, one per split (in order of split_sizes) */
|
|
3761
|
+
readonly Y_splits: ArrayType<ArrayType<ArrayType<FloatType>>>;
|
|
3762
|
+
/** Indices of rows rejected due to rare stratify classes or missing overlap values */
|
|
3863
3763
|
readonly rejected_indices: ArrayType<IntegerType>;
|
|
3864
3764
|
}>;
|
|
3865
3765
|
/** Model blob type for sklearn models */
|