@elaraai/east-py-datascience 0.0.2-beta.8 → 0.0.2-beta.81
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/README.md +58 -1
- package/dist/src/alns/alns.d.ts +528 -0
- package/dist/src/alns/alns.d.ts.map +1 -0
- package/dist/src/alns/alns.js +238 -0
- package/dist/src/alns/alns.js.map +1 -0
- package/dist/src/google_or/google_or.d.ts +2422 -0
- package/dist/src/google_or/google_or.d.ts.map +1 -0
- package/dist/src/google_or/google_or.js +542 -0
- package/dist/src/google_or/google_or.js.map +1 -0
- package/dist/{gp → src/gp}/gp.d.ts +185 -136
- package/dist/src/gp/gp.d.ts.map +1 -0
- package/dist/{gp → src/gp}/gp.js +64 -12
- package/dist/src/gp/gp.js.map +1 -0
- package/dist/src/index.d.ts +34 -0
- package/dist/src/index.d.ts.map +1 -0
- package/dist/src/index.js +57 -0
- package/dist/src/index.js.map +1 -0
- package/dist/src/lightgbm/lightgbm.d.ts +575 -0
- package/dist/src/lightgbm/lightgbm.d.ts.map +1 -0
- package/dist/{lightgbm → src/lightgbm}/lightgbm.js +104 -18
- package/dist/src/lightgbm/lightgbm.js.map +1 -0
- package/dist/src/lightning/lightning.d.ts +1594 -0
- package/dist/src/lightning/lightning.d.ts.map +1 -0
- package/dist/src/lightning/lightning.js +468 -0
- package/dist/src/lightning/lightning.js.map +1 -0
- package/dist/{mads → src/mads}/mads.d.ts +109 -112
- package/dist/src/mads/mads.d.ts.map +1 -0
- package/dist/{mads → src/mads}/mads.js +6 -8
- package/dist/src/mads/mads.js.map +1 -0
- package/dist/src/mapie/mapie.d.ts +3680 -0
- package/dist/src/mapie/mapie.d.ts.map +1 -0
- package/dist/src/mapie/mapie.js +616 -0
- package/dist/src/mapie/mapie.js.map +1 -0
- package/dist/{ngboost → src/ngboost}/ngboost.d.ts +192 -142
- package/dist/src/ngboost/ngboost.d.ts.map +1 -0
- package/dist/{ngboost → src/ngboost}/ngboost.js +67 -14
- package/dist/src/ngboost/ngboost.js.map +1 -0
- package/dist/src/optimization/optimization.d.ts +420 -0
- package/dist/src/optimization/optimization.d.ts.map +1 -0
- package/dist/src/optimization/optimization.js +257 -0
- package/dist/src/optimization/optimization.js.map +1 -0
- package/dist/{optuna → src/optuna}/optuna.d.ts +374 -314
- package/dist/src/optuna/optuna.d.ts.map +1 -0
- package/dist/{optuna → src/optuna}/optuna.js +2 -0
- package/dist/src/optuna/optuna.js.map +1 -0
- package/dist/src/pymc/pymc.d.ts +2932 -0
- package/dist/src/pymc/pymc.d.ts.map +1 -0
- package/dist/src/pymc/pymc.js +688 -0
- package/dist/src/pymc/pymc.js.map +1 -0
- package/dist/src/scipy/scipy.d.ts +2205 -0
- package/dist/src/scipy/scipy.d.ts.map +1 -0
- package/dist/src/scipy/scipy.js +884 -0
- package/dist/src/scipy/scipy.js.map +1 -0
- package/dist/src/shap/shap.d.ts +4611 -0
- package/dist/src/shap/shap.d.ts.map +1 -0
- package/dist/src/shap/shap.js +552 -0
- package/dist/src/shap/shap.js.map +1 -0
- package/dist/{simanneal → src/simanneal}/simanneal.d.ts +257 -160
- package/dist/src/simanneal/simanneal.d.ts.map +1 -0
- package/dist/{simanneal → src/simanneal}/simanneal.js +105 -8
- package/dist/src/simanneal/simanneal.js.map +1 -0
- package/dist/src/simulation/simulation.d.ts +431 -0
- package/dist/src/simulation/simulation.d.ts.map +1 -0
- package/dist/src/simulation/simulation.js +306 -0
- package/dist/src/simulation/simulation.js.map +1 -0
- package/dist/src/sklearn/sklearn.d.ts +6362 -0
- package/dist/src/sklearn/sklearn.d.ts.map +1 -0
- package/dist/src/sklearn/sklearn.js +1508 -0
- package/dist/src/sklearn/sklearn.js.map +1 -0
- package/dist/src/torch/torch.d.ts +1205 -0
- package/dist/src/torch/torch.d.ts.map +1 -0
- package/dist/{torch → src/torch}/torch.js +109 -18
- package/dist/src/torch/torch.js.map +1 -0
- package/dist/src/types.d.ts +43 -0
- package/dist/src/types.d.ts.map +1 -0
- package/dist/src/types.js +44 -0
- package/dist/src/types.js.map +1 -0
- package/dist/src/xgboost/xgboost.d.ts +1424 -0
- package/dist/src/xgboost/xgboost.d.ts.map +1 -0
- package/dist/src/xgboost/xgboost.js +432 -0
- package/dist/src/xgboost/xgboost.js.map +1 -0
- package/package.json +12 -12
- package/dist/gp/gp.d.ts.map +0 -1
- package/dist/gp/gp.js.map +0 -1
- package/dist/index.d.ts +0 -27
- package/dist/index.d.ts.map +0 -1
- package/dist/index.js +0 -41
- package/dist/index.js.map +0 -1
- package/dist/lightgbm/lightgbm.d.ts +0 -494
- package/dist/lightgbm/lightgbm.d.ts.map +0 -1
- package/dist/lightgbm/lightgbm.js.map +0 -1
- package/dist/mads/mads.d.ts.map +0 -1
- package/dist/mads/mads.js.map +0 -1
- package/dist/ngboost/ngboost.d.ts.map +0 -1
- package/dist/ngboost/ngboost.js.map +0 -1
- package/dist/optuna/optuna.d.ts.map +0 -1
- package/dist/optuna/optuna.js.map +0 -1
- package/dist/scipy/scipy.d.ts +0 -1260
- package/dist/scipy/scipy.d.ts.map +0 -1
- package/dist/scipy/scipy.js +0 -413
- package/dist/scipy/scipy.js.map +0 -1
- package/dist/shap/shap.d.ts +0 -657
- package/dist/shap/shap.d.ts.map +0 -1
- package/dist/shap/shap.js +0 -241
- package/dist/shap/shap.js.map +0 -1
- package/dist/simanneal/simanneal.d.ts.map +0 -1
- package/dist/simanneal/simanneal.js.map +0 -1
- package/dist/sklearn/sklearn.d.ts +0 -2691
- package/dist/sklearn/sklearn.d.ts.map +0 -1
- package/dist/sklearn/sklearn.js +0 -524
- package/dist/sklearn/sklearn.js.map +0 -1
- package/dist/torch/torch.d.ts +0 -1081
- package/dist/torch/torch.d.ts.map +0 -1
- package/dist/torch/torch.js.map +0 -1
- package/dist/tsconfig.tsbuildinfo +0 -1
- package/dist/types.d.ts +0 -80
- package/dist/types.d.ts.map +0 -1
- package/dist/types.js +0 -81
- package/dist/types.js.map +0 -1
- package/dist/xgboost/xgboost.d.ts +0 -504
- package/dist/xgboost/xgboost.d.ts.map +0 -1
- package/dist/xgboost/xgboost.js +0 -177
- package/dist/xgboost/xgboost.js.map +0 -1
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"xgboost.d.ts","sourceRoot":"","sources":["../../../src/xgboost/xgboost.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EAEH,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,QAAQ,EACX,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGrD,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAMrD;;GAEG;AACH,eAAO,MAAM,iBAAiB;IAC1B,8CAA8C;;IAE9C,qCAAqC;;IAErC,wDAAwD;;IAExD,mEAAmE;;IAEnE,0DAA0D;;IAE1D,uEAAuE;;IAEvE,yCAAyC;;IAEzC,yCAAyC;;IAEzC,8EAA8E;;IAE9E,sCAAsC;;IAEtC,4DAA4D;;IAE5D,oEAAoE;;IAEpE,mEAAmE;;IAEnE;;;sFAGkF;;IAElF,iHAAiH;;IAEjH,mFAAmF;;EAErF,CAAC;AAEH;;;GAGG;AACH,eAAO,MAAM,yBAAyB;IAClC,6EAA6E;;IAE7E,8CAA8C;;IAE9C,qCAAqC;;IAErC,wDAAwD;;IAExD,mEAAmE;;IAEnE,0DAA0D;;IAE1D,uEAAuE;;IAEvE,yCAAyC;;IAEzC,yCAAyC;;IAEzC,8EAA8E;;IAE9E,sCAAsC;;IAEtC,4DAA4D;;IAE5D,oEAAoE;;IAEpE,mEAAmE;;IAEnE;;;sFAGkF;;IAElF,iHAAiH;;IAEjH,mFAAmF;;EAErF,CAAC;AAMH;;;;GAIG;AACH,eAAO,MAAM,oBAAoB;IAC7B,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;EAGzF,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,gCAAgC;IACzC,0CAA0C;;IAE1C,oDAAoD;;EAEtD,CAAC;AAMH;;;;;;;GAOG;AACH,eAAO,MAAM,uBAAuB;IAxJhC,8CAA8C;;IAE9C,qCAAqC;;IAErC,wDAAwD;;IAExD,mEAAmE;;IAEnE,0DAA0D;;IAE1D,uEAAuE;;IAEvE,yCAAyC;;IAEzC,yCAAyC;;IAEzC,8EAA8E;;IAE9E,sCAAsC;;IAEtC,4DAA4D;;IAE5D,oEAAoE;;IAEpE,mEAAmE;;IAEnE;;;sFAGkF;;IAElF,iHAAiH;;IAEjH,mFAAmF;;;IA0DnF,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;GA+B1F,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,wBAAwB;IAtKjC,8CAA8C;;IAE9C,qCAAqC;;IAErC,wDAAwD;;IAExD,mEAAmE;;IAEnE,0DAA0D;;IAE1D,uEAAuE;;IAEvE,yCAAyC;;IAEzC,yCAAyC;;IAEzC,8EAA8E;;IAE9E,sCAAsC;;IAEtC,4DAA4D;;IAE5D,oEAAoE;;IAEpE,mEAAmE;;IAEnE;;;sFAGkF;;IAElF,iHAAiH;;IAEjH,mFAAmF;;;IA0DnF,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;GA6C1F,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,eAAe;IAxFxB,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;kDA0D1F,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,qBAAqB;IArG9B,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;oDAuE1F,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,qBAAqB;IAlH9B,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;kDAoF1F,CAAC;AAEF;;;;;;;;;;GAUG;AACH,eAAO,MAAM,sBAAsB;IApL/B,6EAA6E;;IAE7E,8CAA8C;;IAE9C,qCAAqC;;IAErC,wDAAwD;;IAExD,mEAAmE;;IAEnE,0DAA0D;;IAE1D,uEAAuE;;IAEvE,yCAAyC;;IAEzC,yCAAyC;;IAEzC,8EAA8E;;IAE9E,sCAAsC;;IAEtC,4DAA4D;;IAE5D,oEAAoE;;IAEpE,mEAAmE;;IAEnE;;;sFAGkF;;IAElF,iHAAiH;;IAEjH,mFAAmF;;;IAcnF,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;GAqG1F,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,wBAAwB;IAhJjC,8BAA8B;;QAE1B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,+BAA+B;;QAE3B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,wBAAwB;;QAExB,8DAA8D;;QAE9D,mFAAmF;;;IAGvF,qEAAqE;;QAEjE,uDAAuD;;QAEvD,oCAAoC;;QAEpC,+BAA+B;;QAE/B,8DAA8D;;QAE9D,mFAAmF;;;;IASvF,0CAA0C;;IAE1C,oDAAoD;;GAuGvD,CAAC;AAMF;;GAEG;AACH,eAAO,MAAM,YAAY;IACrB,iCAAiC;;QAzPjC,8CAA8C;;QAE9C,qCAAqC;;QAErC,wDAAwD;;QAExD,mEAAmE;;QAEnE,0DAA0D;;QAE1D,uEAAuE;;QAEvE,yCAAyC;;QAEzC,yCAAyC;;QAEzC,8EAA8E;;QAE9E,sCAAsC;;QAEtC,4DAA4D;;QAE5D,oEAAoE;;QAEpE,mEAAmE;;QAEnE;;;0FAGkF;;QAElF,iHAAiH;;QAEjH,mFAAmF;;;IA0NnF,0CAA0C;;QAjN1C,6EAA6E;;QAE7E,8CAA8C;;QAE9C,qCAAqC;;QAErC,wDAAwD;;QAExD,mEAAmE;;QAEnE,0DAA0D;;QAE1D,uEAAuE;;QAEvE,yCAAyC;;QAEzC,yCAAyC;;QAEzC,8EAA8E;;QAE9E,sCAAsC;;QAEtC,4DAA4D;;QAE5D,oEAAoE;;QAEpE,mEAAmE;;QAEnE;;;0FAGkF;;QAElF,iHAAiH;;QAEjH,mFAAmF;;;IAgLnF,sCAAsC;;QAvHtC,0CAA0C;;QAE1C,oDAAoD;;;IAuHpD,yCAAyC;;QApKzC,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;;CAoIjF,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,eAAO,MAAM,OAAO;IAChB;;;;;;;;;;;;;;;;;;;;;;;;OAwBG;;QA1TH,8CAA8C;;QAE9C,qCAAqC;;QAErC,wDAAwD;;QAExD,mEAAmE;;QAEnE,0DAA0D;;QAE1D,uEAAuE;;QAEvE,yCAAyC;;QAEzC,yCAAyC;;QAEzC,8EAA8E;;QAE9E,sCAAsC;;QAEtC,4DAA4D;;QAE5D,oEAAoE;;QAEpE,mEAAmE;;QAEnE;;;0FAGkF;;QAElF,iHAAiH;;QAEjH,mFAAmF;;;QA0DnF,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;;IAgMvF;;;;;;;;;;;;;;;;;;;;;;;;OAwBG;;QArVH,8CAA8C;;QAE9C,qCAAqC;;QAErC,wDAAwD;;QAExD,mEAAmE;;QAEnE,0DAA0D;;QAE1D,uEAAuE;;QAEvE,yCAAyC;;QAEzC,yCAAyC;;QAEzC,8EAA8E;;QAE9E,sCAAsC;;QAEtC,4DAA4D;;QAE5D,oEAAoE;;QAEpE,mEAAmE;;QAEnE;;;0FAGkF;;QAElF,iHAAiH;;QAEjH,mFAAmF;;;QA0DnF,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;;IA2NvF;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA4BG;;QA1UH,6EAA6E;;QAE7E,8CAA8C;;QAE9C,qCAAqC;;QAErC,wDAAwD;;QAExD,mEAAmE;;QAEnE,0DAA0D;;QAE1D,uEAAuE;;QAEvE,yCAAyC;;QAEzC,yCAAyC;;QAEzC,8EAA8E;;QAE9E,sCAAsC;;QAEtC,4DAA4D;;QAE5D,oEAAoE;;QAEpE,mEAAmE;;QAEnE;;;0FAGkF;;QAElF,iHAAiH;;QAEjH,mFAAmF;;;QAcnF,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;;IA0PvF;;;;;;;;;;;;;;;;OAgBG;;QA5SH,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;;IA6QvF;;;;;;;;;;;;;;;;OAgBG;;QA/TH,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;;IAgSvF;;;;;;;;;;;;;;;;;OAiBG;;QAnVH,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;;IAoTvF;;;;;;;;;;;;;;;;;;;OAmBG;;QAzWH,8BAA8B;;YAE1B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,+BAA+B;;YAE3B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,wBAAwB;;YAExB,8DAA8D;;YAE9D,mFAAmF;;;QAGvF,qEAAqE;;YAEjE,uDAAuD;;YAEvD,oCAAoC;;YAEpC,+BAA+B;;YAE/B,8DAA8D;;YAE9D,mFAAmF;;;;QASvF,0CAA0C;;QAE1C,oDAAoD;;;IA8TpD,uBAAuB;;QA7MvB,iCAAiC;;YAzPjC,8CAA8C;;YAE9C,qCAAqC;;YAErC,wDAAwD;;YAExD,mEAAmE;;YAEnE,0DAA0D;;YAE1D,uEAAuE;;YAEvE,yCAAyC;;YAEzC,yCAAyC;;YAEzC,8EAA8E;;YAE9E,sCAAsC;;YAEtC,4DAA4D;;YAE5D,oEAAoE;;YAEpE,mEAAmE;;YAEnE;;;8FAGkF;;YAElF,iHAAiH;;YAEjH,mFAAmF;;;QA0NnF,0CAA0C;;YAjN1C,6EAA6E;;YAE7E,8CAA8C;;YAE9C,qCAAqC;;YAErC,wDAAwD;;YAExD,mEAAmE;;YAEnE,0DAA0D;;YAE1D,uEAAuE;;YAEvE,yCAAyC;;YAEzC,yCAAyC;;YAEzC,8EAA8E;;YAE9E,sCAAsC;;YAEtC,4DAA4D;;YAE5D,oEAAoE;;YAEpE,mEAAmE;;YAEnE;;;8FAGkF;;YAElF,iHAAiH;;YAEjH,mFAAmF;;;QAgLnF,sCAAsC;;YAvHtC,0CAA0C;;YAE1C,oDAAoD;;;QAuHpD,yCAAyC;;YApKzC,8BAA8B;;gBAE1B,mCAAmC;;gBAEnC,+BAA+B;;gBAE/B,8DAA8D;;gBAE9D,mFAAmF;;;YAGvF,+BAA+B;;gBAE3B,mCAAmC;;gBAEnC,+BAA+B;;gBAE/B,wBAAwB;;gBAExB,8DAA8D;;gBAE9D,mFAAmF;;;YAGvF,qEAAqE;;gBAEjE,uDAAuD;;gBAEvD,oCAAoC;;gBAEpC,+BAA+B;;gBAE/B,8DAA8D;;gBAE9D,mFAAmF;;;;;CA2UjF,CAAC"}
|
|
@@ -0,0 +1,432 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Copyright (c) 2025 Elara AI Pty Ltd
|
|
3
|
+
* Dual-licensed under AGPL-3.0 and commercial license. See LICENSE for details.
|
|
4
|
+
*/
|
|
5
|
+
/**
|
|
6
|
+
* XGBoost platform functions for East.
|
|
7
|
+
*
|
|
8
|
+
* Provides gradient boosting for regression and classification.
|
|
9
|
+
* Uses ONNX for model serialization to enable portable inference.
|
|
10
|
+
*
|
|
11
|
+
* @packageDocumentation
|
|
12
|
+
*/
|
|
13
|
+
import { East, StructType, VariantType, OptionType, IntegerType, FloatType, BlobType, } from "@elaraai/east";
|
|
14
|
+
import { VectorType, MatrixType } from "../types.js";
|
|
15
|
+
// Re-export shared types for convenience
|
|
16
|
+
export { VectorType, MatrixType } from "../types.js";
|
|
17
|
+
// ============================================================================
|
|
18
|
+
// Config Types
|
|
19
|
+
// ============================================================================
|
|
20
|
+
/**
|
|
21
|
+
* Configuration for XGBoost models.
|
|
22
|
+
*/
|
|
23
|
+
export const XGBoostConfigType = StructType({
|
|
24
|
+
/** Number of boosting rounds (default 100) */
|
|
25
|
+
n_estimators: OptionType(IntegerType),
|
|
26
|
+
/** Maximum tree depth (default 6) */
|
|
27
|
+
max_depth: OptionType(IntegerType),
|
|
28
|
+
/** Learning rate / step size shrinkage (default 0.3) */
|
|
29
|
+
learning_rate: OptionType(FloatType),
|
|
30
|
+
/** Minimum sum of instance weight needed in a child (default 1) */
|
|
31
|
+
min_child_weight: OptionType(IntegerType),
|
|
32
|
+
/** Subsample ratio of training instances (default 1.0) */
|
|
33
|
+
subsample: OptionType(FloatType),
|
|
34
|
+
/** Subsample ratio of columns when constructing trees (default 1.0) */
|
|
35
|
+
colsample_bytree: OptionType(FloatType),
|
|
36
|
+
/** L1 regularization term (default 0) */
|
|
37
|
+
reg_alpha: OptionType(FloatType),
|
|
38
|
+
/** L2 regularization term (default 1) */
|
|
39
|
+
reg_lambda: OptionType(FloatType),
|
|
40
|
+
/** Minimum loss reduction required to make a further partition (default 0) */
|
|
41
|
+
gamma: OptionType(FloatType),
|
|
42
|
+
/** Random seed for reproducibility */
|
|
43
|
+
random_state: OptionType(IntegerType),
|
|
44
|
+
/** Number of parallel threads (default -1 for all cores) */
|
|
45
|
+
n_jobs: OptionType(IntegerType),
|
|
46
|
+
/** Sample weights for training (one per sample, default uniform) */
|
|
47
|
+
sample_weight: OptionType(VectorType(FloatType)),
|
|
48
|
+
/** Column indices that contain categorical features (0-indexed) */
|
|
49
|
+
categorical_features: OptionType(VectorType(IntegerType)),
|
|
50
|
+
/** Number of categories per categorical feature (one per entry in categorical_features).
|
|
51
|
+
* Used to set explicit pandas Categorical range at both train and predict time,
|
|
52
|
+
* ensuring consistent category spaces. Values outside [0, n) become NaN at predict
|
|
53
|
+
* time, which XGBoost handles natively via learned default branch directions. */
|
|
54
|
+
categorical_n: OptionType(VectorType(IntegerType)),
|
|
55
|
+
/** Max categories for one-hot encoding (default 4). Features with more categories use partition-based splits. */
|
|
56
|
+
max_cat_to_onehot: OptionType(IntegerType),
|
|
57
|
+
/** Max categories considered per split for partition-based method (default 64). */
|
|
58
|
+
max_cat_threshold: OptionType(IntegerType),
|
|
59
|
+
});
|
|
60
|
+
/**
|
|
61
|
+
* Configuration for XGBoost quantile regression.
|
|
62
|
+
* Trains separate models for each quantile.
|
|
63
|
+
*/
|
|
64
|
+
export const XGBoostQuantileConfigType = StructType({
|
|
65
|
+
/** Quantiles to predict (e.g., [0.1, 0.5, 0.9] for 80% interval + median) */
|
|
66
|
+
quantiles: VectorType(FloatType),
|
|
67
|
+
/** Number of boosting rounds (default 100) */
|
|
68
|
+
n_estimators: OptionType(IntegerType),
|
|
69
|
+
/** Maximum tree depth (default 6) */
|
|
70
|
+
max_depth: OptionType(IntegerType),
|
|
71
|
+
/** Learning rate / step size shrinkage (default 0.3) */
|
|
72
|
+
learning_rate: OptionType(FloatType),
|
|
73
|
+
/** Minimum sum of instance weight needed in a child (default 1) */
|
|
74
|
+
min_child_weight: OptionType(IntegerType),
|
|
75
|
+
/** Subsample ratio of training instances (default 1.0) */
|
|
76
|
+
subsample: OptionType(FloatType),
|
|
77
|
+
/** Subsample ratio of columns when constructing trees (default 1.0) */
|
|
78
|
+
colsample_bytree: OptionType(FloatType),
|
|
79
|
+
/** L1 regularization term (default 0) */
|
|
80
|
+
reg_alpha: OptionType(FloatType),
|
|
81
|
+
/** L2 regularization term (default 1) */
|
|
82
|
+
reg_lambda: OptionType(FloatType),
|
|
83
|
+
/** Minimum loss reduction required to make a further partition (default 0) */
|
|
84
|
+
gamma: OptionType(FloatType),
|
|
85
|
+
/** Random seed for reproducibility */
|
|
86
|
+
random_state: OptionType(IntegerType),
|
|
87
|
+
/** Number of parallel threads (default -1 for all cores) */
|
|
88
|
+
n_jobs: OptionType(IntegerType),
|
|
89
|
+
/** Sample weights for training (one per sample, default uniform) */
|
|
90
|
+
sample_weight: OptionType(VectorType(FloatType)),
|
|
91
|
+
/** Column indices that contain categorical features (0-indexed) */
|
|
92
|
+
categorical_features: OptionType(VectorType(IntegerType)),
|
|
93
|
+
/** Number of categories per categorical feature (one per entry in categorical_features).
|
|
94
|
+
* Used to set explicit pandas Categorical range at both train and predict time,
|
|
95
|
+
* ensuring consistent category spaces. Values outside [0, n) become NaN at predict
|
|
96
|
+
* time, which XGBoost handles natively via learned default branch directions. */
|
|
97
|
+
categorical_n: OptionType(VectorType(IntegerType)),
|
|
98
|
+
/** Max categories for one-hot encoding (default 4). Features with more categories use partition-based splits. */
|
|
99
|
+
max_cat_to_onehot: OptionType(IntegerType),
|
|
100
|
+
/** Max categories considered per split for partition-based method (default 64). */
|
|
101
|
+
max_cat_threshold: OptionType(IntegerType),
|
|
102
|
+
});
|
|
103
|
+
// ============================================================================
|
|
104
|
+
// Model Blob Types
|
|
105
|
+
// ============================================================================
|
|
106
|
+
/**
|
|
107
|
+
* Model blob type for serialized XGBoost models.
|
|
108
|
+
*
|
|
109
|
+
* Each model type has its own variant case containing cloudpickle bytes and metadata.
|
|
110
|
+
*/
|
|
111
|
+
export const XGBoostModelBlobType = VariantType({
|
|
112
|
+
/** XGBoost regressor model */
|
|
113
|
+
xgboost_regressor: StructType({
|
|
114
|
+
/** Cloudpickle serialized model */
|
|
115
|
+
data: BlobType,
|
|
116
|
+
/** Number of input features */
|
|
117
|
+
n_features: IntegerType,
|
|
118
|
+
/** Column indices of categorical features (for prediction) */
|
|
119
|
+
categorical_features: OptionType(VectorType(IntegerType)),
|
|
120
|
+
/** Number of categories per categorical feature (for consistent category space) */
|
|
121
|
+
categorical_n: OptionType(VectorType(IntegerType)),
|
|
122
|
+
}),
|
|
123
|
+
/** XGBoost classifier model */
|
|
124
|
+
xgboost_classifier: StructType({
|
|
125
|
+
/** Cloudpickle serialized model */
|
|
126
|
+
data: BlobType,
|
|
127
|
+
/** Number of input features */
|
|
128
|
+
n_features: IntegerType,
|
|
129
|
+
/** Number of classes */
|
|
130
|
+
n_classes: IntegerType,
|
|
131
|
+
/** Column indices of categorical features (for prediction) */
|
|
132
|
+
categorical_features: OptionType(VectorType(IntegerType)),
|
|
133
|
+
/** Number of categories per categorical feature (for consistent category space) */
|
|
134
|
+
categorical_n: OptionType(VectorType(IntegerType)),
|
|
135
|
+
}),
|
|
136
|
+
/** XGBoost quantile regressor (multiple models, one per quantile) */
|
|
137
|
+
xgboost_quantile: StructType({
|
|
138
|
+
/** Cloudpickle serialized dict of {quantile: model} */
|
|
139
|
+
data: BlobType,
|
|
140
|
+
/** Quantiles this model predicts */
|
|
141
|
+
quantiles: VectorType(FloatType),
|
|
142
|
+
/** Number of input features */
|
|
143
|
+
n_features: IntegerType,
|
|
144
|
+
/** Column indices of categorical features (for prediction) */
|
|
145
|
+
categorical_features: OptionType(VectorType(IntegerType)),
|
|
146
|
+
/** Number of categories per categorical feature (for consistent category space) */
|
|
147
|
+
categorical_n: OptionType(VectorType(IntegerType)),
|
|
148
|
+
}),
|
|
149
|
+
});
|
|
150
|
+
/**
|
|
151
|
+
* Result from XGBoost quantile prediction.
|
|
152
|
+
*/
|
|
153
|
+
export const XGBoostQuantilePredictResultType = StructType({
|
|
154
|
+
/** Quantile values that were predicted */
|
|
155
|
+
quantiles: VectorType(FloatType),
|
|
156
|
+
/** Predictions matrix: (n_samples x n_quantiles) */
|
|
157
|
+
predictions: MatrixType(FloatType),
|
|
158
|
+
});
|
|
159
|
+
// ============================================================================
|
|
160
|
+
// Platform Functions
|
|
161
|
+
// ============================================================================
|
|
162
|
+
/**
|
|
163
|
+
* Train an XGBoost regression model.
|
|
164
|
+
*
|
|
165
|
+
* @param X - Feature matrix
|
|
166
|
+
* @param y - Target vector
|
|
167
|
+
* @param config - XGBoost configuration
|
|
168
|
+
* @returns Model blob containing trained regressor
|
|
169
|
+
*/
|
|
170
|
+
export const xgboost_train_regressor = East.platform("xgboost_train_regressor", [MatrixType(FloatType), VectorType(FloatType), XGBoostConfigType], XGBoostModelBlobType);
|
|
171
|
+
/**
|
|
172
|
+
* Train an XGBoost classification model.
|
|
173
|
+
*
|
|
174
|
+
* @param X - Feature matrix
|
|
175
|
+
* @param y - Label vector (integer class labels)
|
|
176
|
+
* @param config - XGBoost configuration
|
|
177
|
+
* @returns Model blob containing trained classifier
|
|
178
|
+
*/
|
|
179
|
+
export const xgboost_train_classifier = East.platform("xgboost_train_classifier", [MatrixType(FloatType), VectorType(IntegerType), XGBoostConfigType], XGBoostModelBlobType);
|
|
180
|
+
/**
|
|
181
|
+
* Make predictions with a trained XGBoost regressor.
|
|
182
|
+
*
|
|
183
|
+
* @param model - Trained regressor model blob
|
|
184
|
+
* @param X - Feature matrix
|
|
185
|
+
* @returns Predicted values
|
|
186
|
+
*/
|
|
187
|
+
export const xgboost_predict = East.platform("xgboost_predict", [XGBoostModelBlobType, MatrixType(FloatType)], VectorType(FloatType));
|
|
188
|
+
/**
|
|
189
|
+
* Predict class labels with a trained XGBoost classifier.
|
|
190
|
+
*
|
|
191
|
+
* @param model - Trained classifier model blob
|
|
192
|
+
* @param X - Feature matrix
|
|
193
|
+
* @returns Predicted class labels
|
|
194
|
+
*/
|
|
195
|
+
export const xgboost_predict_class = East.platform("xgboost_predict_class", [XGBoostModelBlobType, MatrixType(FloatType)], VectorType(IntegerType));
|
|
196
|
+
/**
|
|
197
|
+
* Get class probabilities from a trained XGBoost classifier.
|
|
198
|
+
*
|
|
199
|
+
* @param model - Trained classifier model blob
|
|
200
|
+
* @param X - Feature matrix
|
|
201
|
+
* @returns Probability matrix (n_samples x n_classes)
|
|
202
|
+
*/
|
|
203
|
+
export const xgboost_predict_proba = East.platform("xgboost_predict_proba", [XGBoostModelBlobType, MatrixType(FloatType)], MatrixType(FloatType));
|
|
204
|
+
/**
|
|
205
|
+
* Train XGBoost quantile regression models.
|
|
206
|
+
*
|
|
207
|
+
* Trains one model per quantile using pinball loss (reg:quantileerror).
|
|
208
|
+
* This provides prediction intervals and uncertainty quantification.
|
|
209
|
+
*
|
|
210
|
+
* @param X - Feature matrix
|
|
211
|
+
* @param y - Target vector
|
|
212
|
+
* @param config - Quantile regression configuration (includes quantiles array)
|
|
213
|
+
* @returns Model blob containing trained quantile models
|
|
214
|
+
*/
|
|
215
|
+
export const xgboost_train_quantile = East.platform("xgboost_train_quantile", [MatrixType(FloatType), VectorType(FloatType), XGBoostQuantileConfigType], XGBoostModelBlobType);
|
|
216
|
+
/**
|
|
217
|
+
* Predict quantiles with trained XGBoost quantile regressor.
|
|
218
|
+
*
|
|
219
|
+
* @param model - Trained quantile model blob
|
|
220
|
+
* @param X - Feature matrix
|
|
221
|
+
* @returns Quantiles and predictions matrix (n_samples x n_quantiles)
|
|
222
|
+
*/
|
|
223
|
+
export const xgboost_predict_quantile = East.platform("xgboost_predict_quantile", [XGBoostModelBlobType, MatrixType(FloatType)], XGBoostQuantilePredictResultType);
|
|
224
|
+
// ============================================================================
|
|
225
|
+
// Grouped Export
|
|
226
|
+
// ============================================================================
|
|
227
|
+
/**
|
|
228
|
+
* Type definitions for XGBoost functions.
|
|
229
|
+
*/
|
|
230
|
+
export const XGBoostTypes = {
|
|
231
|
+
/** XGBoost configuration type */
|
|
232
|
+
XGBoostConfigType,
|
|
233
|
+
/** XGBoost quantile configuration type */
|
|
234
|
+
XGBoostQuantileConfigType,
|
|
235
|
+
/** Quantile prediction result type */
|
|
236
|
+
XGBoostQuantilePredictResultType,
|
|
237
|
+
/** Model blob type for XGBoost models */
|
|
238
|
+
ModelBlobType: XGBoostModelBlobType,
|
|
239
|
+
};
|
|
240
|
+
/**
|
|
241
|
+
* XGBoost gradient boosting.
|
|
242
|
+
*
|
|
243
|
+
* Provides regression and classification with gradient boosted decision trees.
|
|
244
|
+
*
|
|
245
|
+
* @example
|
|
246
|
+
* ```ts
|
|
247
|
+
* import { East, variant } from "@elaraai/east";
|
|
248
|
+
* import { XGBoost } from "@elaraai/east-py-datascience";
|
|
249
|
+
*
|
|
250
|
+
* const train = East.function([], XGBoost.Types.ModelBlobType, $ => {
|
|
251
|
+
* const X = $.let([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0], [7.0, 8.0]]);
|
|
252
|
+
* const y = $.let([1.0, 2.0, 3.0, 4.0]);
|
|
253
|
+
* const config = $.let({
|
|
254
|
+
* n_estimators: variant('some', 100n),
|
|
255
|
+
* max_depth: variant('some', 3n),
|
|
256
|
+
* learning_rate: variant('some', 0.1),
|
|
257
|
+
* min_child_weight: variant('none', null),
|
|
258
|
+
* subsample: variant('none', null),
|
|
259
|
+
* colsample_bytree: variant('none', null),
|
|
260
|
+
* reg_alpha: variant('none', null),
|
|
261
|
+
* reg_lambda: variant('none', null),
|
|
262
|
+
* random_state: variant('some', 42n),
|
|
263
|
+
* n_jobs: variant('none', null),
|
|
264
|
+
* sample_weight: variant('none', null),
|
|
265
|
+
* });
|
|
266
|
+
* return $.return(XGBoost.trainRegressor(X, y, config));
|
|
267
|
+
* });
|
|
268
|
+
* ```
|
|
269
|
+
*/
|
|
270
|
+
export const XGBoost = {
|
|
271
|
+
/**
|
|
272
|
+
* Train an XGBoost regression model.
|
|
273
|
+
*
|
|
274
|
+
* @example
|
|
275
|
+
* ```ts
|
|
276
|
+
* import { East, FloatType, MatrixType, VectorType, variant } from "@elaraai/east";
|
|
277
|
+
* import { XGBoost, XGBoostConfigType } from "@elaraai/east-py-datascience";
|
|
278
|
+
*
|
|
279
|
+
* const train = East.function([], XGBoost.Types.ModelBlobType, ($) => {
|
|
280
|
+
* const X = $.let([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0], [7.0, 8.0]]);
|
|
281
|
+
* const y = $.let(new Float64Array([1.0, 2.0, 3.0, 4.0]));
|
|
282
|
+
* const config = $.let({
|
|
283
|
+
* n_estimators: variant("some", 100n), max_depth: variant("some", 3n),
|
|
284
|
+
* learning_rate: variant("some", 0.1), min_child_weight: variant("none", null),
|
|
285
|
+
* subsample: variant("none", null), colsample_bytree: variant("none", null),
|
|
286
|
+
* reg_alpha: variant("none", null), reg_lambda: variant("none", null),
|
|
287
|
+
* gamma: variant("none", null), random_state: variant("some", 42n),
|
|
288
|
+
* n_jobs: variant("none", null), sample_weight: variant("none", null),
|
|
289
|
+
* categorical_features: variant("none", null), categorical_n: variant("none", null),
|
|
290
|
+
* max_cat_to_onehot: variant("none", null), max_cat_threshold: variant("none", null),
|
|
291
|
+
* }, XGBoostConfigType);
|
|
292
|
+
* return $.return(XGBoost.trainRegressor(X, y, config));
|
|
293
|
+
* });
|
|
294
|
+
* ```
|
|
295
|
+
*/
|
|
296
|
+
trainRegressor: xgboost_train_regressor,
|
|
297
|
+
/**
|
|
298
|
+
* Train an XGBoost classification model.
|
|
299
|
+
*
|
|
300
|
+
* @example
|
|
301
|
+
* ```ts
|
|
302
|
+
* import { East, FloatType, IntegerType, MatrixType, VectorType, variant } from "@elaraai/east";
|
|
303
|
+
* import { XGBoost, XGBoostConfigType } from "@elaraai/east-py-datascience";
|
|
304
|
+
*
|
|
305
|
+
* const train = East.function([], XGBoost.Types.ModelBlobType, ($) => {
|
|
306
|
+
* const X = $.let([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0], [7.0, 8.0]]);
|
|
307
|
+
* const y = $.let(new BigInt64Array([0n, 1n, 0n, 1n]));
|
|
308
|
+
* const config = $.let({
|
|
309
|
+
* n_estimators: variant("some", 50n), max_depth: variant("some", 3n),
|
|
310
|
+
* learning_rate: variant("some", 0.1), min_child_weight: variant("none", null),
|
|
311
|
+
* subsample: variant("none", null), colsample_bytree: variant("none", null),
|
|
312
|
+
* reg_alpha: variant("none", null), reg_lambda: variant("none", null),
|
|
313
|
+
* gamma: variant("none", null), random_state: variant("some", 42n),
|
|
314
|
+
* n_jobs: variant("none", null), sample_weight: variant("none", null),
|
|
315
|
+
* categorical_features: variant("none", null), categorical_n: variant("none", null),
|
|
316
|
+
* max_cat_to_onehot: variant("none", null), max_cat_threshold: variant("none", null),
|
|
317
|
+
* }, XGBoostConfigType);
|
|
318
|
+
* return $.return(XGBoost.trainClassifier(X, y, config));
|
|
319
|
+
* });
|
|
320
|
+
* ```
|
|
321
|
+
*/
|
|
322
|
+
trainClassifier: xgboost_train_classifier,
|
|
323
|
+
/**
|
|
324
|
+
* Train XGBoost quantile regression models.
|
|
325
|
+
*
|
|
326
|
+
* Trains one model per quantile using pinball loss, providing
|
|
327
|
+
* prediction intervals and uncertainty quantification.
|
|
328
|
+
*
|
|
329
|
+
* @example
|
|
330
|
+
* ```ts
|
|
331
|
+
* import { East, FloatType, MatrixType, VectorType, variant } from "@elaraai/east";
|
|
332
|
+
* import { XGBoost, XGBoostQuantileConfigType } from "@elaraai/east-py-datascience";
|
|
333
|
+
*
|
|
334
|
+
* const train = East.function([], XGBoost.Types.ModelBlobType, ($) => {
|
|
335
|
+
* const X = $.let([[1.0], [2.0], [3.0], [4.0]]);
|
|
336
|
+
* const y = $.let(new Float64Array([1.0, 2.0, 3.0, 4.0]));
|
|
337
|
+
* const config = $.let({
|
|
338
|
+
* quantiles: new Float64Array([0.1, 0.5, 0.9]),
|
|
339
|
+
* n_estimators: variant("some", 100n), max_depth: variant("some", 3n),
|
|
340
|
+
* learning_rate: variant("some", 0.1), min_child_weight: variant("none", null),
|
|
341
|
+
* subsample: variant("none", null), colsample_bytree: variant("none", null),
|
|
342
|
+
* reg_alpha: variant("none", null), reg_lambda: variant("none", null),
|
|
343
|
+
* gamma: variant("none", null), random_state: variant("some", 42n),
|
|
344
|
+
* n_jobs: variant("none", null), sample_weight: variant("none", null),
|
|
345
|
+
* categorical_features: variant("none", null), categorical_n: variant("none", null),
|
|
346
|
+
* max_cat_to_onehot: variant("none", null), max_cat_threshold: variant("none", null),
|
|
347
|
+
* }, XGBoostQuantileConfigType);
|
|
348
|
+
* return $.return(XGBoost.trainQuantile(X, y, config));
|
|
349
|
+
* });
|
|
350
|
+
* ```
|
|
351
|
+
*/
|
|
352
|
+
trainQuantile: xgboost_train_quantile,
|
|
353
|
+
/**
|
|
354
|
+
* Make predictions with a trained XGBoost regressor.
|
|
355
|
+
*
|
|
356
|
+
* @example
|
|
357
|
+
* ```ts
|
|
358
|
+
* import { East, FloatType, MatrixType, VectorType } from "@elaraai/east";
|
|
359
|
+
* import { XGBoost } from "@elaraai/east-py-datascience";
|
|
360
|
+
*
|
|
361
|
+
* const predictFn = East.function(
|
|
362
|
+
* [XGBoost.Types.ModelBlobType, MatrixType(FloatType)],
|
|
363
|
+
* VectorType(FloatType),
|
|
364
|
+
* ($, model, X) => {
|
|
365
|
+
* return $.return(XGBoost.predict(model, X));
|
|
366
|
+
* }
|
|
367
|
+
* );
|
|
368
|
+
* ```
|
|
369
|
+
*/
|
|
370
|
+
predict: xgboost_predict,
|
|
371
|
+
/**
|
|
372
|
+
* Predict class labels with a trained XGBoost classifier.
|
|
373
|
+
*
|
|
374
|
+
* @example
|
|
375
|
+
* ```ts
|
|
376
|
+
* import { East, FloatType, IntegerType, MatrixType, VectorType } from "@elaraai/east";
|
|
377
|
+
* import { XGBoost } from "@elaraai/east-py-datascience";
|
|
378
|
+
*
|
|
379
|
+
* const predictFn = East.function(
|
|
380
|
+
* [XGBoost.Types.ModelBlobType, MatrixType(FloatType)],
|
|
381
|
+
* VectorType(IntegerType),
|
|
382
|
+
* ($, model, X) => {
|
|
383
|
+
* return $.return(XGBoost.predictClass(model, X));
|
|
384
|
+
* }
|
|
385
|
+
* );
|
|
386
|
+
* ```
|
|
387
|
+
*/
|
|
388
|
+
predictClass: xgboost_predict_class,
|
|
389
|
+
/**
|
|
390
|
+
* Get class probabilities from a trained XGBoost classifier.
|
|
391
|
+
*
|
|
392
|
+
* @example
|
|
393
|
+
* ```ts
|
|
394
|
+
* import { East, FloatType, MatrixType } from "@elaraai/east";
|
|
395
|
+
* import { XGBoost } from "@elaraai/east-py-datascience";
|
|
396
|
+
*
|
|
397
|
+
* const predictFn = East.function(
|
|
398
|
+
* [XGBoost.Types.ModelBlobType, MatrixType(FloatType)],
|
|
399
|
+
* MatrixType(FloatType),
|
|
400
|
+
* ($, model, X) => {
|
|
401
|
+
* // Returns (n_samples x n_classes) probability matrix
|
|
402
|
+
* return $.return(XGBoost.predictProba(model, X));
|
|
403
|
+
* }
|
|
404
|
+
* );
|
|
405
|
+
* ```
|
|
406
|
+
*/
|
|
407
|
+
predictProba: xgboost_predict_proba,
|
|
408
|
+
/**
|
|
409
|
+
* Predict quantiles with a trained XGBoost quantile regressor.
|
|
410
|
+
*
|
|
411
|
+
* @example
|
|
412
|
+
* ```ts
|
|
413
|
+
* import { East, FloatType, MatrixType } from "@elaraai/east";
|
|
414
|
+
* import { XGBoost } from "@elaraai/east-py-datascience";
|
|
415
|
+
*
|
|
416
|
+
* const predictFn = East.function(
|
|
417
|
+
* [XGBoost.Types.ModelBlobType, MatrixType(FloatType)],
|
|
418
|
+
* XGBoost.Types.XGBoostQuantilePredictResultType,
|
|
419
|
+
* ($, model, X) => {
|
|
420
|
+
* const result = $.let(XGBoost.predictQuantile(model, X));
|
|
421
|
+
* // result.quantiles => [0.1, 0.5, 0.9]
|
|
422
|
+
* // result.predictions => (n_samples x n_quantiles) matrix
|
|
423
|
+
* return $.return(result);
|
|
424
|
+
* }
|
|
425
|
+
* );
|
|
426
|
+
* ```
|
|
427
|
+
*/
|
|
428
|
+
predictQuantile: xgboost_predict_quantile,
|
|
429
|
+
/** Type definitions */
|
|
430
|
+
Types: XGBoostTypes,
|
|
431
|
+
};
|
|
432
|
+
//# sourceMappingURL=xgboost.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"xgboost.js","sourceRoot":"","sources":["../../../src/xgboost/xgboost.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EACH,IAAI,EACJ,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,QAAQ,GACX,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,yCAAyC;AACzC,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,UAAU,CAAC;IACxC,8CAA8C;IAC9C,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,qCAAqC;IACrC,SAAS,EAAE,UAAU,CAAC,WAAW,CAAC;IAClC,wDAAwD;IACxD,aAAa,EAAE,UAAU,CAAC,SAAS,CAAC;IACpC,mEAAmE;IACnE,gBAAgB,EAAE,UAAU,CAAC,WAAW,CAAC;IACzC,0DAA0D;IAC1D,SAAS,EAAE,UAAU,CAAC,SAAS,CAAC;IAChC,uEAAuE;IACvE,gBAAgB,EAAE,UAAU,CAAC,SAAS,CAAC;IACvC,yCAAyC;IACzC,SAAS,EAAE,UAAU,CAAC,SAAS,CAAC;IAChC,yCAAyC;IACzC,UAAU,EAAE,UAAU,CAAC,SAAS,CAAC;IACjC,8EAA8E;IAC9E,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC;IAC5B,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,4DAA4D;IAC5D,MAAM,EAAE,UAAU,CAAC,WAAW,CAAC;IAC/B,oEAAoE;IACpE,aAAa,EAAE,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;IAChD,mEAAmE;IACnE,oBAAoB,EAAE,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;IACzD;;;sFAGkF;IAClF,aAAa,EAAE,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;IAClD,iHAAiH;IACjH,iBAAiB,EAAE,UAAU,CAAC,WAAW,CAAC;IAC1C,mFAAmF;IACnF,iBAAiB,EAAE,UAAU,CAAC,WAAW,CAAC;CAC7C,CAAC,CAAC;AAEH;;;GAGG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,UAAU,CAAC;IAChD,6EAA6E;IAC7E,SAAS,EAAE,UAAU,CAAC,SAAS,CAAC;IAChC,8CAA8C;IAC9C,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,qCAAqC;IACrC,SAAS,EAAE,UAAU,CAAC,WAAW,CAAC;IAClC,wDAAwD;IACxD,aAAa,EAAE,UAAU,CAAC,SAAS,CAAC;IACpC,mEAAmE;IACnE,gBAAgB,EAAE,UAAU,CAAC,WAAW,CAAC;IACzC,0DAA0D;IAC1D,SAAS,EAAE,UAAU,CAAC,SAAS,CAAC;IAChC,uEAAuE;IACvE,gBAAgB,EAAE,UAAU,CAAC,SAAS,CAAC;IACvC,yCAAyC;IACzC,SAAS,EAAE,UAAU,CAAC,SAAS,CAAC;IAChC,yCAAyC;IACzC,UAAU,EAAE,UAAU,CAAC,SAAS,CAAC;IACjC,8EAA8E;IAC9E,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC;IAC5B,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,4DAA4D;IAC5D,MAAM,EAAE,UAAU,CAAC,WAAW,CAAC;IAC/B,oEAAoE;IACpE,aAAa,EAAE,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;IAChD,mEAAmE;IACnE,oBAAoB,EAAE,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;IACzD;;;sFAGkF;IAClF,aAAa,EAAE,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;IAClD,iHAAiH;IACjH,iBAAiB,EAAE,UAAU,CAAC,WAAW,CAAC;IAC1C,mFAAmF;IACnF,iBAAiB,EAAE,UAAU,CAAC,WAAW,CAAC;CAC7C,CAAC,CAAC;AAEH,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;GAIG;AACH,MAAM,CAAC,MAAM,oBAAoB,GAAG,WAAW,CAAC;IAC5C,8BAA8B;IAC9B,iBAAiB,EAAE,UAAU,CAAC;QAC1B,mCAAmC;QACnC,IAAI,EAAE,QAAQ;QACd,+BAA+B;QAC/B,UAAU,EAAE,WAAW;QACvB,8DAA8D;QAC9D,oBAAoB,EAAE,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QACzD,mFAAmF;QACnF,aAAa,EAAE,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;KACrD,CAAC;IACF,+BAA+B;IAC/B,kBAAkB,EAAE,UAAU,CAAC;QAC3B,mCAAmC;QACnC,IAAI,EAAE,QAAQ;QACd,+BAA+B;QAC/B,UAAU,EAAE,WAAW;QACvB,wBAAwB;QACxB,SAAS,EAAE,WAAW;QACtB,8DAA8D;QAC9D,oBAAoB,EAAE,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QACzD,mFAAmF;QACnF,aAAa,EAAE,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;KACrD,CAAC;IACF,qEAAqE;IACrE,gBAAgB,EAAE,UAAU,CAAC;QACzB,uDAAuD;QACvD,IAAI,EAAE,QAAQ;QACd,oCAAoC;QACpC,SAAS,EAAE,UAAU,CAAC,SAAS,CAAC;QAChC,+BAA+B;QAC/B,UAAU,EAAE,WAAW;QACvB,8DAA8D;QAC9D,oBAAoB,EAAE,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;QACzD,mFAAmF;QACnF,aAAa,EAAE,UAAU,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC;KACrD,CAAC;CACL,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,gCAAgC,GAAG,UAAU,CAAC;IACvD,0CAA0C;IAC1C,SAAS,EAAE,UAAU,CAAC,SAAS,CAAC;IAChC,oDAAoD;IACpD,WAAW,EAAE,UAAU,CAAC,SAAS,CAAC;CACrC,CAAC,CAAC;AAEH,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAE/E;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,IAAI,CAAC,QAAQ,CAChD,yBAAyB,EACzB,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,iBAAiB,CAAC,EACjE,oBAAoB,CACvB,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,wBAAwB,GAAG,IAAI,CAAC,QAAQ,CACjD,0BAA0B,EAC1B,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,WAAW,CAAC,EAAE,iBAAiB,CAAC,EACnE,oBAAoB,CACvB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CACxC,iBAAiB,EACjB,CAAC,oBAAoB,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,EAC7C,UAAU,CAAC,SAAS,CAAC,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,IAAI,CAAC,QAAQ,CAC9C,uBAAuB,EACvB,CAAC,oBAAoB,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,EAC7C,UAAU,CAAC,WAAW,CAAC,CAC1B,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,IAAI,CAAC,QAAQ,CAC9C,uBAAuB,EACvB,CAAC,oBAAoB,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,EAC7C,UAAU,CAAC,SAAS,CAAC,CACxB,CAAC;AAEF;;;;;;;;;;GAUG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG,IAAI,CAAC,QAAQ,CAC/C,wBAAwB,EACxB,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,yBAAyB,CAAC,EACzE,oBAAoB,CACvB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,wBAAwB,GAAG,IAAI,CAAC,QAAQ,CACjD,0BAA0B,EAC1B,CAAC,oBAAoB,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,EAC7C,gCAAgC,CACnC,CAAC;AAEF,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG;IACxB,iCAAiC;IACjC,iBAAiB;IACjB,0CAA0C;IAC1C,yBAAyB;IACzB,sCAAsC;IACtC,gCAAgC;IAChC,yCAAyC;IACzC,aAAa,EAAE,oBAAoB;CAC7B,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,MAAM,CAAC,MAAM,OAAO,GAAG;IACnB;;;;;;;;;;;;;;;;;;;;;;;;OAwBG;IACH,cAAc,EAAE,uBAAuB;IAEvC;;;;;;;;;;;;;;;;;;;;;;;;OAwBG;IACH,eAAe,EAAE,wBAAwB;IAEzC;;;;;;;;;;;;;;;;;;;;;;;;;;;;OA4BG;IACH,aAAa,EAAE,sBAAsB;IAErC;;;;;;;;;;;;;;;;OAgBG;IACH,OAAO,EAAE,eAAe;IAExB;;;;;;;;;;;;;;;;OAgBG;IACH,YAAY,EAAE,qBAAqB;IAEnC;;;;;;;;;;;;;;;;;OAiBG;IACH,YAAY,EAAE,qBAAqB;IAEnC;;;;;;;;;;;;;;;;;;;OAmBG;IACH,eAAe,EAAE,wBAAwB;IACzC,uBAAuB;IACvB,KAAK,EAAE,YAAY;CACb,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,17 +1,17 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@elaraai/east-py-datascience",
|
|
3
|
-
"version": "0.0.2-beta.
|
|
3
|
+
"version": "0.0.2-beta.81",
|
|
4
4
|
"description": "East Data Science - ML/optimization platform functions for East (TypeScript definitions)",
|
|
5
|
-
"main": "dist/index.js",
|
|
6
|
-
"types": "dist/index.d.ts",
|
|
5
|
+
"main": "dist/src/index.js",
|
|
6
|
+
"types": "dist/src/index.d.ts",
|
|
7
7
|
"exports": {
|
|
8
8
|
".": {
|
|
9
|
-
"types": "./dist/index.d.ts",
|
|
10
|
-
"default": "./dist/index.js"
|
|
9
|
+
"types": "./dist/src/index.d.ts",
|
|
10
|
+
"default": "./dist/src/index.js"
|
|
11
11
|
}
|
|
12
12
|
},
|
|
13
13
|
"files": [
|
|
14
|
-
"dist",
|
|
14
|
+
"dist/src",
|
|
15
15
|
"!dist/**/*.spec.*",
|
|
16
16
|
"README.md",
|
|
17
17
|
"LICENSE.md",
|
|
@@ -20,9 +20,9 @@
|
|
|
20
20
|
],
|
|
21
21
|
"scripts": {
|
|
22
22
|
"build": "tsc",
|
|
23
|
-
"test": "npm run build && node --enable-source-maps --test 'dist/**/*.spec.js'",
|
|
24
|
-
"test:coverage": "npm run build && node --enable-source-maps --test --experimental-test-coverage 'dist/**/*.spec.js'",
|
|
25
|
-
"test:export": "npm run build && EXPORT_TEST_IR=/tmp/east-py-datascience
|
|
23
|
+
"test": "npm run build && node --enable-source-maps --test 'dist/test/**/*.spec.js' 'dist/src/**/*.spec.js'",
|
|
24
|
+
"test:coverage": "npm run build && node --enable-source-maps --test --experimental-test-coverage 'dist/test/**/*.spec.js' 'dist/src/**/*.spec.js'",
|
|
25
|
+
"test:export": "npm run build && EXPORT_TEST_IR=/tmp/east-py-datascience node --enable-source-maps --test 'dist/test/**/*.spec.js'",
|
|
26
26
|
"lint": "eslint .",
|
|
27
27
|
"lint:fix": "eslint . --fix",
|
|
28
28
|
"version:patch": "npm version patch -m 'chore: bump version to %s'",
|
|
@@ -75,8 +75,8 @@
|
|
|
75
75
|
"eslint-plugin-headers": "^1.3.3",
|
|
76
76
|
"typescript": "~5.9.2"
|
|
77
77
|
},
|
|
78
|
-
"
|
|
79
|
-
"@elaraai/east": "
|
|
80
|
-
"@elaraai/east-node-std": "
|
|
78
|
+
"peerDependencies": {
|
|
79
|
+
"@elaraai/east": "beta",
|
|
80
|
+
"@elaraai/east-node-std": "beta"
|
|
81
81
|
}
|
|
82
82
|
}
|
package/dist/gp/gp.d.ts.map
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"gp.d.ts","sourceRoot":"","sources":["../../src/gp/gp.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EAEH,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,WAAW,EACX,QAAQ,EACR,UAAU,EACV,QAAQ,EACX,MAAM,eAAe,CAAC;AAIvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAMrD;;GAEG;AACH,eAAO,MAAM,YAAY;IACrB,kDAAkD;;IAElD,uCAAuC;;IAEvC,yBAAyB;;IAEzB,yBAAyB;;IAEzB,yBAAyB;;IAEzB,2BAA2B;;EAE7B,CAAC;AAMH;;GAEG;AACH,eAAO,MAAM,YAAY;IACrB,gCAAgC;;QAtBhC,kDAAkD;;QAElD,uCAAuC;;QAEvC,yBAAyB;;QAEzB,yBAAyB;;QAEzB,yBAAyB;;QAEzB,2BAA2B;;;IAc3B,oDAAoD;;IAEpD,mDAAmD;;IAEnD,yDAAyD;;IAEzD,sCAAsC;;EAExC,CAAC;AAMH;;GAEG;AACH,eAAO,MAAM,mBAAmB;IAC5B,4BAA4B;;IAE5B,iDAAiD;;EAEnD,CAAC;AAMH;;GAEG;AACH,eAAO,MAAM,eAAe;IACxB,iCAAiC;;QAE7B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,qCAAqC;;;EAG3C,CAAC;AAMH;;;;;;;GAOG;AACH,eAAO,MAAM,QAAQ;IAzDjB,gCAAgC;;QAtBhC,kDAAkD;;QAElD,uCAAuC;;QAEvC,yBAAyB;;QAEzB,yBAAyB;;QAEzB,yBAAyB;;QAEzB,2BAA2B;;;IAc3B,oDAAoD;;IAEpD,mDAAmD;;IAEnD,yDAAyD;;IAEzD,sCAAsC;;;IA0BtC,iCAAiC;;QAE7B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,qCAAqC;;;GAqB5C,CAAC;AAEF;;;;;;;;GAQG;AACH,eAAO,MAAM,UAAU;IAtCnB,iCAAiC;;QAE7B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,qCAAqC;;;mIAoC5C,CAAC;AAEF;;;;;;;;GAQG;AACH,eAAO,MAAM,cAAc;IArDvB,iCAAiC;;QAE7B,mCAAmC;;QAEnC,+BAA+B;;QAE/B,qCAAqC;;;;IApBzC,4BAA4B;;IAE5B,iDAAiD;;GAqEpD,CAAC;AAMF;;GAEG;AACH,eAAO,MAAM,OAAO;IAChB,oCAAoC;;IAEpC,uCAAuC;;IAEvC,kBAAkB;;QA/HlB,kDAAkD;;QAElD,uCAAuC;;QAEvC,yBAAyB;;QAEzB,yBAAyB;;QAEzB,yBAAyB;;QAEzB,2BAA2B;;;IAuH3B,yBAAyB;;QA3GzB,gCAAgC;;YAtBhC,kDAAkD;;YAElD,uCAAuC;;YAEvC,yBAAyB;;YAEzB,yBAAyB;;YAEzB,yBAAyB;;YAEzB,2BAA2B;;;QAc3B,oDAAoD;;QAEpD,mDAAmD;;QAEnD,yDAAyD;;QAEzD,sCAAsC;;;IAqGtC,8CAA8C;;QAzF9C,4BAA4B;;QAE5B,iDAAiD;;;IAyFjD,oCAAoC;;QA7EpC,iCAAiC;;YAE7B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,qCAAqC;;;;CAyEnC,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,eAAO,MAAM,EAAE;IACX,yBAAyB;;QA5IzB,gCAAgC;;YAtBhC,kDAAkD;;YAElD,uCAAuC;;YAEvC,yBAAyB;;YAEzB,yBAAyB;;YAEzB,yBAAyB;;YAEzB,2BAA2B;;;QAc3B,oDAAoD;;QAEpD,mDAAmD;;QAEnD,yDAAyD;;QAEzD,sCAAsC;;;QA0BtC,iCAAiC;;YAE7B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,qCAAqC;;;;IAsGzC,mCAAmC;;QA5GnC,iCAAiC;;YAE7B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,qCAAqC;;;;IAwGzC,wCAAwC;;QA9GxC,iCAAiC;;YAE7B,mCAAmC;;YAEnC,+BAA+B;;YAE/B,qCAAqC;;;;QApBzC,4BAA4B;;QAE5B,iDAAiD;;;IA4HjD,uBAAuB;;QA7CvB,oCAAoC;;QAEpC,uCAAuC;;QAEvC,kBAAkB;;YA/HlB,kDAAkD;;YAElD,uCAAuC;;YAEvC,yBAAyB;;YAEzB,yBAAyB;;YAEzB,yBAAyB;;YAEzB,2BAA2B;;;QAuH3B,yBAAyB;;YA3GzB,gCAAgC;;gBAtBhC,kDAAkD;;gBAElD,uCAAuC;;gBAEvC,yBAAyB;;gBAEzB,yBAAyB;;gBAEzB,yBAAyB;;gBAEzB,2BAA2B;;;YAc3B,oDAAoD;;YAEpD,mDAAmD;;YAEnD,yDAAyD;;YAEzD,sCAAsC;;;QAqGtC,8CAA8C;;YAzF9C,4BAA4B;;YAE5B,iDAAiD;;;QAyFjD,oCAAoC;;YA7EpC,iCAAiC;;gBAE7B,mCAAmC;;gBAEnC,+BAA+B;;gBAE/B,qCAAqC;;;;;CA4GnC,CAAC"}
|
package/dist/gp/gp.js.map
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"gp.js","sourceRoot":"","sources":["../../src/gp/gp.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;GAOG;AAEH,OAAO,EACH,IAAI,EACJ,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,WAAW,EACX,QAAQ,EACR,UAAU,EACV,QAAQ,GACX,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,yCAAyC;AACzC,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,+EAA+E;AAC/E,aAAa;AACb,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG,WAAW,CAAC;IACpC,kDAAkD;IAClD,GAAG,EAAE,QAAQ;IACb,uCAAuC;IACvC,UAAU,EAAE,QAAQ;IACpB,yBAAyB;IACzB,UAAU,EAAE,QAAQ;IACpB,yBAAyB;IACzB,UAAU,EAAE,QAAQ;IACpB,yBAAyB;IACzB,kBAAkB,EAAE,QAAQ;IAC5B,2BAA2B;IAC3B,WAAW,EAAE,QAAQ;CACxB,CAAC,CAAC;AAEH,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG,UAAU,CAAC;IACnC,gCAAgC;IAChC,MAAM,EAAE,UAAU,CAAC,YAAY,CAAC;IAChC,oDAAoD;IACpD,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC;IAC5B,mDAAmD;IACnD,oBAAoB,EAAE,UAAU,CAAC,WAAW,CAAC;IAC7C,yDAAyD;IACzD,WAAW,EAAE,UAAU,CAAC,WAAW,CAAC;IACpC,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;CACxC,CAAC,CAAC;AAEH,+EAA+E;AAC/E,eAAe;AACf,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,UAAU,CAAC;IAC1C,4BAA4B;IAC5B,IAAI,EAAE,UAAU;IAChB,iDAAiD;IACjD,GAAG,EAAE,UAAU;CAClB,CAAC,CAAC;AAEH,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,WAAW,CAAC;IACvC,iCAAiC;IACjC,YAAY,EAAE,UAAU,CAAC;QACrB,mCAAmC;QACnC,IAAI,EAAE,QAAQ;QACd,+BAA+B;QAC/B,UAAU,EAAE,WAAW;QACvB,qCAAqC;QACrC,WAAW,EAAE,UAAU;KAC1B,CAAC;CACL,CAAC,CAAC;AAEH,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAE/E;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CACjC,UAAU,EACV,CAAC,UAAU,EAAE,UAAU,EAAE,YAAY,CAAC,EACtC,eAAe,CAClB,CAAC;AAEF;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,UAAU,GAAG,IAAI,CAAC,QAAQ,CACnC,YAAY,EACZ,CAAC,eAAe,EAAE,UAAU,CAAC,EAC7B,UAAU,CACb,CAAC;AAEF;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG,IAAI,CAAC,QAAQ,CACvC,gBAAgB,EAChB,CAAC,eAAe,EAAE,UAAU,CAAC,EAC7B,mBAAmB,CACtB,CAAC;AAEF,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,OAAO,GAAG;IACnB,oCAAoC;IACpC,UAAU;IACV,uCAAuC;IACvC,UAAU;IACV,kBAAkB;IAClB,YAAY;IACZ,yBAAyB;IACzB,YAAY;IACZ,8CAA8C;IAC9C,mBAAmB;IACnB,oCAAoC;IACpC,aAAa,EAAE,eAAe;CACxB,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAM,CAAC,MAAM,EAAE,GAAG;IACd,yBAAyB;IACzB,KAAK,EAAE,QAAQ;IACf,mCAAmC;IACnC,OAAO,EAAE,UAAU;IACnB,wCAAwC;IACxC,UAAU,EAAE,cAAc;IAC1B,uBAAuB;IACvB,KAAK,EAAE,OAAO;CACR,CAAC"}
|
package/dist/index.d.ts
DELETED
|
@@ -1,27 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Copyright (c) 2025 Elara AI Pty Ltd
|
|
3
|
-
* Dual-licensed under AGPL-3.0 and commercial license. See LICENSE for details.
|
|
4
|
-
*/
|
|
5
|
-
/**
|
|
6
|
-
* East Data Science - ML and optimization platform functions for East.
|
|
7
|
-
*
|
|
8
|
-
* This package provides data science capabilities for East programs:
|
|
9
|
-
* - MADS: Derivative-free blackbox optimization (PyNomadBBO)
|
|
10
|
-
* - Optuna: Bayesian optimization with TPE sampler
|
|
11
|
-
* - SimAnneal: Discrete optimization with Simulated Annealing
|
|
12
|
-
*
|
|
13
|
-
* @packageDocumentation
|
|
14
|
-
*/
|
|
15
|
-
export { MADS, mads_optimize, MADSTypes, VectorType, ScalarObjectiveType, MADSBoundsType, MADSConstraintType, MADSDirectionType, MADSConfigType, MADSResultType, } from "./mads/mads.js";
|
|
16
|
-
export { Optuna, optuna_optimize, OptunaTypes, ParamValueType, ParamSpaceKindType, ParamSpaceType, NamedParamType, OptimizationDirectionType, PrunerType, OptunaStudyConfigType, TrialResultType, StudyResultType, ObjectiveFunctionType, } from "./optuna/optuna.js";
|
|
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";
|
|
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
|
-
export { XGBoost, xgboost_train_regressor, xgboost_train_classifier, xgboost_predict, xgboost_predict_class, xgboost_predict_proba, XGBoostTypes, XGBoostConfigType, XGBoostModelBlobType, } from "./xgboost/xgboost.js";
|
|
21
|
-
export { LightGBM, lightgbm_train_regressor, lightgbm_train_classifier, lightgbm_predict, lightgbm_predict_class, lightgbm_predict_proba, LightGBMTypes, LightGBMConfigType, LightGBMModelBlobType, } from "./lightgbm/lightgbm.js";
|
|
22
|
-
export { NGBoost, ngboost_train_regressor, ngboost_predict, ngboost_predict_dist, NGBoostTypes, NGBoostDistributionType, NGBoostConfigType, NGBoostPredictConfigType, NGBoostPredictResultType, NGBoostModelBlobType, } from "./ngboost/ngboost.js";
|
|
23
|
-
export { Shap, shap_tree_explainer_create, shap_kernel_explainer_create, shap_compute_values, shap_feature_importance, ShapTypes, ShapResultType, FeatureImportanceType, ShapModelBlobType, AnyModelBlobType, StringVectorType, } from "./shap/shap.js";
|
|
24
|
-
export { Torch, torch_mlp_train, torch_mlp_predict, TorchTypes, TorchActivationType, TorchLossType, TorchOptimizerType, TorchMLPConfigType, TorchTrainConfigType, TorchTrainResultType, TorchTrainOutputType, TorchModelBlobType, } from "./torch/torch.js";
|
|
25
|
-
export { GP, gp_train, gp_predict, gp_predict_std, GPTypes, GPKernelType, GPConfigType, GPPredictResultType, GPModelBlobType, } from "./gp/gp.js";
|
|
26
|
-
export { VectorType as SharedVectorType, MatrixType as SharedMatrixType, ScalarObjectiveType as SharedScalarObjectiveType, VectorObjectiveType, LabelVectorType, } from "./types.js";
|
|
27
|
-
//# sourceMappingURL=index.d.ts.map
|
package/dist/index.d.ts.map
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;;;GASG;AAGH,OAAO,EACH,IAAI,EACJ,aAAa,EACb,SAAS,EACT,UAAU,EACV,mBAAmB,EACnB,cAAc,EACd,kBAAkB,EAClB,iBAAiB,EACjB,cAAc,EACd,cAAc,GACjB,MAAM,gBAAgB,CAAC;AAGxB,OAAO,EACH,MAAM,EACN,eAAe,EACf,WAAW,EACX,cAAc,EACd,kBAAkB,EAClB,cAAc,EACd,cAAc,EACd,yBAAyB,EACzB,UAAU,EACV,qBAAqB,EACrB,eAAe,EACf,eAAe,EACf,qBAAqB,GACxB,MAAM,oBAAoB,CAAC;AAG5B,OAAO,EACH,SAAS,EACT,kBAAkB,EAClB,8BAA8B,EAC9B,yBAAyB,EACzB,cAAc,EACd,iBAAiB,EACjB,kBAAkB,EAClB,gBAAgB,EAChB,qBAAqB,EACrB,gBAAgB,EAChB,gBAAgB,EAChB,gBAAgB,GACnB,MAAM,0BAA0B,CAAC;AAGlC,OAAO,EACH,OAAO,EACP,wBAAwB,EACxB,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,eAAe,EACf,qBAAqB,EACrB,qBAAqB,EACrB,YAAY,EACZ,iBAAiB,EACjB,oBAAoB,GACvB,MAAM,sBAAsB,CAAC;AAG9B,OAAO,EACH,QAAQ,EACR,wBAAwB,EACxB,yBAAyB,EACzB,gBAAgB,EAChB,sBAAsB,EACtB,sBAAsB,EACtB,aAAa,EACb,kBAAkB,EAClB,qBAAqB,GACxB,MAAM,wBAAwB,CAAC;AAGhC,OAAO,EACH,OAAO,EACP,uBAAuB,EACvB,eAAe,EACf,oBAAoB,EACpB,YAAY,EACZ,uBAAuB,EACvB,iBAAiB,EACjB,wBAAwB,EACxB,wBAAwB,EACxB,oBAAoB,GACvB,MAAM,sBAAsB,CAAC;AAG9B,OAAO,EACH,IAAI,EACJ,0BAA0B,EAC1B,4BAA4B,EAC5B,mBAAmB,EACnB,uBAAuB,EACvB,SAAS,EACT,cAAc,EACd,qBAAqB,EACrB,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,GACnB,MAAM,gBAAgB,CAAC;AAGxB,OAAO,EACH,KAAK,EACL,eAAe,EACf,iBAAiB,EACjB,UAAU,EACV,mBAAmB,EACnB,aAAa,EACb,kBAAkB,EAClB,kBAAkB,EAClB,oBAAoB,EACpB,oBAAoB,EACpB,oBAAoB,EACpB,kBAAkB,GACrB,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EACH,EAAE,EACF,QAAQ,EACR,UAAU,EACV,cAAc,EACd,OAAO,EACP,YAAY,EACZ,YAAY,EACZ,mBAAmB,EACnB,eAAe,GAClB,MAAM,YAAY,CAAC;AAGpB,OAAO,EACH,UAAU,IAAI,gBAAgB,EAC9B,UAAU,IAAI,gBAAgB,EAC9B,mBAAmB,IAAI,yBAAyB,EAChD,mBAAmB,EACnB,eAAe,GAClB,MAAM,YAAY,CAAC"}
|
package/dist/index.js
DELETED
|
@@ -1,41 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Copyright (c) 2025 Elara AI Pty Ltd
|
|
3
|
-
* Dual-licensed under AGPL-3.0 and commercial license. See LICENSE for details.
|
|
4
|
-
*/
|
|
5
|
-
/**
|
|
6
|
-
* East Data Science - ML and optimization platform functions for East.
|
|
7
|
-
*
|
|
8
|
-
* This package provides data science capabilities for East programs:
|
|
9
|
-
* - MADS: Derivative-free blackbox optimization (PyNomadBBO)
|
|
10
|
-
* - Optuna: Bayesian optimization with TPE sampler
|
|
11
|
-
* - SimAnneal: Discrete optimization with Simulated Annealing
|
|
12
|
-
*
|
|
13
|
-
* @packageDocumentation
|
|
14
|
-
*/
|
|
15
|
-
// MADS - Derivative-free optimization
|
|
16
|
-
export { MADS, mads_optimize, MADSTypes, VectorType, ScalarObjectiveType, MADSBoundsType, MADSConstraintType, MADSDirectionType, MADSConfigType, MADSResultType, } from "./mads/mads.js";
|
|
17
|
-
// Optuna - Bayesian optimization
|
|
18
|
-
export { Optuna, optuna_optimize, OptunaTypes, ParamValueType, ParamSpaceKindType, ParamSpaceType, NamedParamType, OptimizationDirectionType, PrunerType, OptunaStudyConfigType, TrialResultType, StudyResultType, ObjectiveFunctionType, } from "./optuna/optuna.js";
|
|
19
|
-
// SimAnneal - Discrete optimization
|
|
20
|
-
export { SimAnneal, simanneal_optimize, simanneal_optimize_permutation, simanneal_optimize_subset, SimAnnealTypes, DiscreteStateType, EnergyFunctionType, MoveFunctionType, PermutationEnergyType, SubsetEnergyType, AnnealConfigType, AnnealResultType, } from "./simanneal/simanneal.js";
|
|
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,
|
|
23
|
-
// Flexible metrics types
|
|
24
|
-
RegressionMetricType, MetricResultType, MetricsResultType, MetricAggregationType, MultiMetricsConfigType, MultiMetricResultType, MultiMetricsResultType, ClassificationMetricType, ClassificationAverageType, ClassificationMetricsConfigType, ClassificationMetricResultType, ClassificationMetricResultsType, MultiClassificationConfigType, MultiClassificationMetricResultType, MultiClassificationMetricResultsType, } from "./sklearn/sklearn.js";
|
|
25
|
-
// Scipy - Scientific computing
|
|
26
|
-
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";
|
|
27
|
-
// XGBoost - Gradient boosting
|
|
28
|
-
export { XGBoost, xgboost_train_regressor, xgboost_train_classifier, xgboost_predict, xgboost_predict_class, xgboost_predict_proba, XGBoostTypes, XGBoostConfigType, XGBoostModelBlobType, } from "./xgboost/xgboost.js";
|
|
29
|
-
// LightGBM - Fast gradient boosting
|
|
30
|
-
export { LightGBM, lightgbm_train_regressor, lightgbm_train_classifier, lightgbm_predict, lightgbm_predict_class, lightgbm_predict_proba, LightGBMTypes, LightGBMConfigType, LightGBMModelBlobType, } from "./lightgbm/lightgbm.js";
|
|
31
|
-
// NGBoost - Probabilistic gradient boosting
|
|
32
|
-
export { NGBoost, ngboost_train_regressor, ngboost_predict, ngboost_predict_dist, NGBoostTypes, NGBoostDistributionType, NGBoostConfigType, NGBoostPredictConfigType, NGBoostPredictResultType, NGBoostModelBlobType, } from "./ngboost/ngboost.js";
|
|
33
|
-
// SHAP - Model explainability
|
|
34
|
-
export { Shap, shap_tree_explainer_create, shap_kernel_explainer_create, shap_compute_values, shap_feature_importance, ShapTypes, ShapResultType, FeatureImportanceType, ShapModelBlobType, AnyModelBlobType, StringVectorType, } from "./shap/shap.js";
|
|
35
|
-
// Torch - PyTorch neural networks
|
|
36
|
-
export { Torch, torch_mlp_train, torch_mlp_predict, TorchTypes, TorchActivationType, TorchLossType, TorchOptimizerType, TorchMLPConfigType, TorchTrainConfigType, TorchTrainResultType, TorchTrainOutputType, TorchModelBlobType, } from "./torch/torch.js";
|
|
37
|
-
// GP - Gaussian Process regression
|
|
38
|
-
export { GP, gp_train, gp_predict, gp_predict_std, GPTypes, GPKernelType, GPConfigType, GPPredictResultType, GPModelBlobType, } from "./gp/gp.js";
|
|
39
|
-
// Shared types
|
|
40
|
-
export { VectorType as SharedVectorType, MatrixType as SharedMatrixType, ScalarObjectiveType as SharedScalarObjectiveType, VectorObjectiveType, LabelVectorType, } from "./types.js";
|
|
41
|
-
//# sourceMappingURL=index.js.map
|
package/dist/index.js.map
DELETED
|
@@ -1 +0,0 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;;;GASG;AAEH,sCAAsC;AACtC,OAAO,EACH,IAAI,EACJ,aAAa,EACb,SAAS,EACT,UAAU,EACV,mBAAmB,EACnB,cAAc,EACd,kBAAkB,EAClB,iBAAiB,EACjB,cAAc,EACd,cAAc,GACjB,MAAM,gBAAgB,CAAC;AAExB,iCAAiC;AACjC,OAAO,EACH,MAAM,EACN,eAAe,EACf,WAAW,EACX,cAAc,EACd,kBAAkB,EAClB,cAAc,EACd,cAAc,EACd,yBAAyB,EACzB,UAAU,EACV,qBAAqB,EACrB,eAAe,EACf,eAAe,EACf,qBAAqB,GACxB,MAAM,oBAAoB,CAAC;AAE5B,oCAAoC;AACpC,OAAO,EACH,SAAS,EACT,kBAAkB,EAClB,8BAA8B,EAC9B,yBAAyB,EACzB,cAAc,EACd,iBAAiB,EACjB,kBAAkB,EAClB,gBAAgB,EAChB,qBAAqB,EACrB,gBAAgB,EAChB,gBAAgB,EAChB,gBAAgB,GACnB,MAAM,0BAA0B,CAAC;AAElC,yBAAyB;AACzB,OAAO,EACH,OAAO,EACP,wBAAwB,EACxB,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,eAAe,EACf,qBAAqB,EACrB,qBAAqB,EACrB,YAAY,EACZ,iBAAiB,EACjB,oBAAoB,GACvB,MAAM,sBAAsB,CAAC;AAE9B,oCAAoC;AACpC,OAAO,EACH,QAAQ,EACR,wBAAwB,EACxB,yBAAyB,EACzB,gBAAgB,EAChB,sBAAsB,EACtB,sBAAsB,EACtB,aAAa,EACb,kBAAkB,EAClB,qBAAqB,GACxB,MAAM,wBAAwB,CAAC;AAEhC,4CAA4C;AAC5C,OAAO,EACH,OAAO,EACP,uBAAuB,EACvB,eAAe,EACf,oBAAoB,EACpB,YAAY,EACZ,uBAAuB,EACvB,iBAAiB,EACjB,wBAAwB,EACxB,wBAAwB,EACxB,oBAAoB,GACvB,MAAM,sBAAsB,CAAC;AAE9B,8BAA8B;AAC9B,OAAO,EACH,IAAI,EACJ,0BAA0B,EAC1B,4BAA4B,EAC5B,mBAAmB,EACnB,uBAAuB,EACvB,SAAS,EACT,cAAc,EACd,qBAAqB,EACrB,iBAAiB,EACjB,gBAAgB,EAChB,gBAAgB,GACnB,MAAM,gBAAgB,CAAC;AAExB,kCAAkC;AAClC,OAAO,EACH,KAAK,EACL,eAAe,EACf,iBAAiB,EACjB,UAAU,EACV,mBAAmB,EACnB,aAAa,EACb,kBAAkB,EAClB,kBAAkB,EAClB,oBAAoB,EACpB,oBAAoB,EACpB,oBAAoB,EACpB,kBAAkB,GACrB,MAAM,kBAAkB,CAAC;AAE1B,mCAAmC;AACnC,OAAO,EACH,EAAE,EACF,QAAQ,EACR,UAAU,EACV,cAAc,EACd,OAAO,EACP,YAAY,EACZ,YAAY,EACZ,mBAAmB,EACnB,eAAe,GAClB,MAAM,YAAY,CAAC;AAEpB,eAAe;AACf,OAAO,EACH,UAAU,IAAI,gBAAgB,EAC9B,UAAU,IAAI,gBAAgB,EAC9B,mBAAmB,IAAI,yBAAyB,EAChD,mBAAmB,EACnB,eAAe,GAClB,MAAM,YAAY,CAAC"}
|