@elaraai/east-py-datascience 0.0.2-beta.8 → 0.0.2-beta.80
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 +2988 -0
- package/dist/src/shap/shap.d.ts.map +1 -0
- package/dist/src/shap/shap.js +500 -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":"lightning.d.ts","sourceRoot":"","sources":["../../../src/lightning/lightning.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;;GAQG;AAEH,OAAO,EAEH,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,QAAQ,EACR,SAAS,EACT,QAAQ,EACR,WAAW,EACX,YAAY,EACZ,UAAU,EACb,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,mBAAmB;IAC5B,0CAA0C;;IAE1C,2CAA2C;;QAEvC,yEAAyE;;;IAG7E,wDAAwD;;QAEpD,wBAAwB;;QAExB,iCAAiC;;;IAGrC,+DAA+D;;QAE3D,4CAA4C;;QAE5C,sCAAsC;;QAEtC,yDAAyD;;;EAG/D,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,QAAQ;;;EAGnB,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,yBAAyB;IAClC,0CAA0C;;QAEtC,yBAAyB;;;IAG7B,+DAA+D;;QAE3D,iCAAiC;;QAEjC,oCAAoC;;QAEpC,iCAAiC;;;IAGrC,iEAAiE;;QAE7D,gDAAgD;;QAEhD,mCAAmC;;QAEnC,+BAA+B;;QAE/B,iDAAiD;;QAEjD,wCAAwC;;QAExC,8DAA8D;;;IAGlE,mEAAmE;;QAE/D,gDAAgD;;QAEhD,mCAAmC;;QAEnC,sBAAsB;;QAEtB,2BAA2B;;QAE3B,6BAA6B;;;;;QAE7B,iDAAiD;;QAEjD,+DAA+D;;QAE/D,8DAA8D;;;IAGlE,oEAAoE;;QAEhE,gDAAgD;;QAEhD,mCAAmC;;QAEnC,sBAAsB;;QAEtB,6DAA6D;;QAE7D,mCAAmC;;QAEnC,mDAAmD;;QAEnD,4CAA4C;;QAE5C,8DAA8D;;;EAGpE,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,0BAA0B,6DAGtC,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,mBAAmB;IAC5B,yBAAyB;;QAjFzB,0CAA0C;;YAEtC,yBAAyB;;;QAG7B,+DAA+D;;YAE3D,iCAAiC;;YAEjC,oCAAoC;;YAEpC,iCAAiC;;;QAGrC,iEAAiE;;YAE7D,gDAAgD;;YAEhD,mCAAmC;;YAEnC,+BAA+B;;YAE/B,iDAAiD;;YAEjD,wCAAwC;;YAExC,8DAA8D;;;QAGlE,mEAAmE;;YAE/D,gDAAgD;;YAEhD,mCAAmC;;YAEnC,sBAAsB;;YAEtB,2BAA2B;;YAE3B,6BAA6B;;;;;YAE7B,iDAAiD;;YAEjD,+DAA+D;;YAE/D,8DAA8D;;;QAGlE,oEAAoE;;YAEhE,gDAAgD;;YAEhD,mCAAmC;;YAEnC,sBAAsB;;YAEtB,6DAA6D;;YAE7D,mCAAmC;;YAEnC,mDAAmD;;YAEnD,4CAA4C;;YAE5C,8DAA8D;;;;IAmBlE,6CAA6C;;QAxH7C,0CAA0C;;QAE1C,2CAA2C;;YAEvC,yEAAyE;;;QAG7E,wDAAwD;;YAEpD,wBAAwB;;YAExB,iCAAiC;;;QAGrC,+DAA+D;;YAE3D,4CAA4C;;YAE5C,sCAAsC;;YAEtC,yDAAyD;;;;IAsG7D,oCAAoC;;IAEpC,oCAAoC;;IAEpC,4CAA4C;;IAE5C,+BAA+B;;IAE/B,kCAAkC;;IAElC,6CAA6C;;IAE7C,kDAAkD;;IAElD,sCAAsC;;IAEtC,0CAA0C;;EAE5C,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,sBAAsB;;QAE3B,mDAAmD;;QAEnD,sBAAsB;;QAEtB,uBAAuB;;QAEvB,wBAAwB;;QAExB,kBAAkB;;QAElB,0CAA0C;;;EAGhD,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,mBAAmB;IAC5B,yBAAyB;;;YAnBrB,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;IAW9C,0BAA0B;;IAE1B,4BAA4B;;IAE5B,sCAAsC;;EAExC,CAAC;AAEH;;GAEG;AACH,eAAO,MAAM,gBAAgB,8CAA+C,CAAC;AAE7E;;;;;GAKG;AACH,eAAO,MAAM,gBAAgB;IACzB,uDAAuD;;QAEnD,qEAAqE;;QAErE,mFAAmF;;;IAGvF,0CAA0C;;EAE5C,CAAC;AAMH;;;;;;;;;;GAUG;AACH,eAAO,MAAM,eAAe;IAhGxB,yBAAyB;;QAjFzB,0CAA0C;;YAEtC,yBAAyB;;;QAG7B,+DAA+D;;YAE3D,iCAAiC;;YAEjC,oCAAoC;;YAEpC,iCAAiC;;;QAGrC,iEAAiE;;YAE7D,gDAAgD;;YAEhD,mCAAmC;;YAEnC,+BAA+B;;YAE/B,iDAAiD;;YAEjD,wCAAwC;;YAExC,8DAA8D;;;QAGlE,mEAAmE;;YAE/D,gDAAgD;;YAEhD,mCAAmC;;YAEnC,sBAAsB;;YAEtB,2BAA2B;;YAE3B,6BAA6B;;;;;YAE7B,iDAAiD;;YAEjD,+DAA+D;;YAE/D,8DAA8D;;;QAGlE,oEAAoE;;YAEhE,gDAAgD;;YAEhD,mCAAmC;;YAEnC,sBAAsB;;YAEtB,6DAA6D;;YAE7D,mCAAmC;;YAEnC,mDAAmD;;YAEnD,4CAA4C;;YAE5C,8DAA8D;;;;IAmBlE,6CAA6C;;QAxH7C,0CAA0C;;QAE1C,2CAA2C;;YAEvC,yEAAyE;;;QAG7E,wDAAwD;;YAEpD,wBAAwB;;YAExB,iCAAiC;;;QAGrC,+DAA+D;;YAE3D,4CAA4C;;YAE5C,sCAAsC;;YAEtC,yDAAyD;;;;IAsG7D,oCAAoC;;IAEpC,oCAAoC;;IAEpC,4CAA4C;;IAE5C,+BAA+B;;IAE/B,kCAAkC;;IAElC,6CAA6C;;IAE7C,kDAAkD;;IAElD,sCAAsC;;IAEtC,0CAA0C;;;IAkD1C,uDAAuD;;QAEnD,qEAAqE;;QAErE,mFAAmF;;;IAGvF,0CAA0C;;;IA7B1C,yBAAyB;;;YAnBrB,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;IAW9C,0BAA0B;;IAE1B,4BAA4B;;IAE5B,sCAAsC;;GA8CzC,CAAC;AAEF;;;;;;;;GAQG;AACH,eAAO,MAAM,iBAAiB;;QAlFtB,mDAAmD;;QAEnD,sBAAsB;;QAEtB,uBAAuB;;QAEvB,wBAAwB;;QAExB,kBAAkB;;QAElB,0CAA0C;;;+IA4EjD,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,gBAAgB;;QA/FrB,mDAAmD;;QAEnD,sBAAsB;;QAEtB,uBAAuB;;QAEvB,wBAAwB;;QAExB,kBAAkB;;QAElB,0CAA0C;;;kDAyFjD,CAAC;AAEF;;;;;;GAMG;AACH,eAAO,MAAM,gBAAgB;;QA5GrB,mDAAmD;;QAEnD,sBAAsB;;QAEtB,uBAAuB;;QAEvB,wBAAwB;;QAExB,kBAAkB;;QAElB,0CAA0C;;;kDAsGjD,CAAC;AAEF;;;;;;;GAOG;AACH,eAAO,MAAM,4BAA4B;;QA1HjC,mDAAmD;;QAEnD,sBAAsB;;QAEtB,uBAAuB;;QAEvB,wBAAwB;;QAExB,kBAAkB;;QAElB,0CAA0C;;;yEAoHjD,CAAC;AAEF;;GAEG;AACH,eAAO,MAAM,2BAA2B;IACpC,kCAAkC;;IAElC,gEAAgE;;IAEhE,+DAA+D;;EAEjE,CAAC;AAEH;;;;;;;;;;;;;GAaG;AACH,eAAO,MAAM,2BAA2B;;QA1JhC,mDAAmD;;QAEnD,sBAAsB;;QAEtB,uBAAuB;;QAEvB,wBAAwB;;QAExB,kBAAkB;;QAElB,0CAA0C;;;;IA0H9C,kCAAkC;;IAElC,gEAAgE;;IAEhE,+DAA+D;;2BAsBlE,CAAC;AAMF;;GAEG;AACH,eAAO,MAAM,cAAc;;QA1TvB,0CAA0C;;QAE1C,2CAA2C;;YAEvC,yEAAyE;;;QAG7E,wDAAwD;;YAEpD,wBAAwB;;YAExB,iCAAiC;;;QAGrC,+DAA+D;;YAE3D,4CAA4C;;YAE5C,sCAAsC;;YAEtC,yDAAyD;;;;;QAiB7D,0CAA0C;;YAEtC,yBAAyB;;;QAG7B,+DAA+D;;YAE3D,iCAAiC;;YAEjC,oCAAoC;;YAEpC,iCAAiC;;;QAGrC,iEAAiE;;YAE7D,gDAAgD;;YAEhD,mCAAmC;;YAEnC,+BAA+B;;YAE/B,iDAAiD;;YAEjD,wCAAwC;;YAExC,8DAA8D;;;QAGlE,mEAAmE;;YAE/D,gDAAgD;;YAEhD,mCAAmC;;YAEnC,sBAAsB;;YAEtB,2BAA2B;;YAE3B,6BAA6B;;;;;YAE7B,iDAAiD;;YAEjD,+DAA+D;;YAE/D,8DAA8D;;;QAGlE,oEAAoE;;YAEhE,gDAAgD;;YAEhD,mCAAmC;;YAEnC,sBAAsB;;YAEtB,6DAA6D;;YAE7D,mCAAmC;;YAEnC,mDAAmD;;YAEnD,4CAA4C;;YAE5C,8DAA8D;;;;;;;;;;QAiBlE,yBAAyB;;YAjFzB,0CAA0C;;gBAEtC,yBAAyB;;;YAG7B,+DAA+D;;gBAE3D,iCAAiC;;gBAEjC,oCAAoC;;gBAEpC,iCAAiC;;;YAGrC,iEAAiE;;gBAE7D,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,+BAA+B;;gBAE/B,iDAAiD;;gBAEjD,wCAAwC;;gBAExC,8DAA8D;;;YAGlE,mEAAmE;;gBAE/D,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,sBAAsB;;gBAEtB,2BAA2B;;gBAE3B,6BAA6B;;;;;gBAE7B,iDAAiD;;gBAEjD,+DAA+D;;gBAE/D,8DAA8D;;;YAGlE,oEAAoE;;gBAEhE,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,sBAAsB;;gBAEtB,6DAA6D;;gBAE7D,mCAAmC;;gBAEnC,mDAAmD;;gBAEnD,4CAA4C;;gBAE5C,8DAA8D;;;;QAmBlE,6CAA6C;;YAxH7C,0CAA0C;;YAE1C,2CAA2C;;gBAEvC,yEAAyE;;;YAG7E,wDAAwD;;gBAEpD,wBAAwB;;gBAExB,iCAAiC;;;YAGrC,+DAA+D;;gBAE3D,4CAA4C;;gBAE5C,sCAAsC;;gBAEtC,yDAAyD;;;;QAsG7D,oCAAoC;;QAEpC,oCAAoC;;QAEpC,4CAA4C;;QAE5C,+BAA+B;;QAE/B,kCAAkC;;QAElC,6CAA6C;;QAE7C,kDAAkD;;QAElD,sCAAsC;;QAEtC,0CAA0C;;;;QA4B1C,yBAAyB;;;gBAnBrB,mDAAmD;;gBAEnD,sBAAsB;;gBAEtB,uBAAuB;;gBAEvB,wBAAwB;;gBAExB,kBAAkB;;gBAElB,0CAA0C;;;;QAW9C,0BAA0B;;QAE1B,4BAA4B;;QAE5B,sCAAsC;;;;;YAzBlC,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;;;QA+B9C,uDAAuD;;YAEnD,qEAAqE;;YAErE,mFAAmF;;;QAGvF,0CAA0C;;;;QAoF1C,kCAAkC;;QAElC,gEAAgE;;QAEhE,+DAA+D;;;CA0CzD,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,eAAO,MAAM,SAAS;IAClB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAiCG;;QA3QH,yBAAyB;;YAjFzB,0CAA0C;;gBAEtC,yBAAyB;;;YAG7B,+DAA+D;;gBAE3D,iCAAiC;;gBAEjC,oCAAoC;;gBAEpC,iCAAiC;;;YAGrC,iEAAiE;;gBAE7D,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,+BAA+B;;gBAE/B,iDAAiD;;gBAEjD,wCAAwC;;gBAExC,8DAA8D;;;YAGlE,mEAAmE;;gBAE/D,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,sBAAsB;;gBAEtB,2BAA2B;;gBAE3B,6BAA6B;;;;;gBAE7B,iDAAiD;;gBAEjD,+DAA+D;;gBAE/D,8DAA8D;;;YAGlE,oEAAoE;;gBAEhE,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,sBAAsB;;gBAEtB,6DAA6D;;gBAE7D,mCAAmC;;gBAEnC,mDAAmD;;gBAEnD,4CAA4C;;gBAE5C,8DAA8D;;;;QAmBlE,6CAA6C;;YAxH7C,0CAA0C;;YAE1C,2CAA2C;;gBAEvC,yEAAyE;;;YAG7E,wDAAwD;;gBAEpD,wBAAwB;;gBAExB,iCAAiC;;;YAGrC,+DAA+D;;gBAE3D,4CAA4C;;gBAE5C,sCAAsC;;gBAEtC,yDAAyD;;;;QAsG7D,oCAAoC;;QAEpC,oCAAoC;;QAEpC,4CAA4C;;QAE5C,+BAA+B;;QAE/B,kCAAkC;;QAElC,6CAA6C;;QAE7C,kDAAkD;;QAElD,sCAAsC;;QAEtC,0CAA0C;;;QAkD1C,uDAAuD;;YAEnD,qEAAqE;;YAErE,mFAAmF;;;QAGvF,0CAA0C;;;QA7B1C,yBAAyB;;;gBAnBrB,mDAAmD;;gBAEnD,sBAAsB;;gBAEtB,uBAAuB;;gBAEvB,wBAAwB;;gBAExB,kBAAkB;;gBAElB,0CAA0C;;;;QAW9C,0BAA0B;;QAE1B,4BAA4B;;QAE5B,sCAAsC;;;IAwNtC;;;;;;;;;;;;;;;;;;OAkBG;;;YAnQC,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;IA4P9C;;;;;;;;;;;;;;;;;OAiBG;;;YAvRC,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;IAgR9C;;;;;;;;;;;;;;;;;OAiBG;;;YA3SC,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;IAoS9C;;;;;;;;;;;;;;;;;OAiBG;;;YA/TC,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;IAwT9C;;;;;;;;;;;;;;;;;;OAkBG;;;YApVC,mDAAmD;;YAEnD,sBAAsB;;YAEtB,uBAAuB;;YAEvB,wBAAwB;;YAExB,kBAAkB;;YAElB,0CAA0C;;;;QA0H9C,kCAAkC;;QAElC,gEAAgE;;QAEhE,+DAA+D;;;IA+M/D;;OAEG;;;YA5eH,0CAA0C;;YAE1C,2CAA2C;;gBAEvC,yEAAyE;;;YAG7E,wDAAwD;;gBAEpD,wBAAwB;;gBAExB,iCAAiC;;;YAGrC,+DAA+D;;gBAE3D,4CAA4C;;gBAE5C,sCAAsC;;gBAEtC,yDAAyD;;;;;YAiB7D,0CAA0C;;gBAEtC,yBAAyB;;;YAG7B,+DAA+D;;gBAE3D,iCAAiC;;gBAEjC,oCAAoC;;gBAEpC,iCAAiC;;;YAGrC,iEAAiE;;gBAE7D,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,+BAA+B;;gBAE/B,iDAAiD;;gBAEjD,wCAAwC;;gBAExC,8DAA8D;;;YAGlE,mEAAmE;;gBAE/D,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,sBAAsB;;gBAEtB,2BAA2B;;gBAE3B,6BAA6B;;;;;gBAE7B,iDAAiD;;gBAEjD,+DAA+D;;gBAE/D,8DAA8D;;;YAGlE,oEAAoE;;gBAEhE,gDAAgD;;gBAEhD,mCAAmC;;gBAEnC,sBAAsB;;gBAEtB,6DAA6D;;gBAE7D,mCAAmC;;gBAEnC,mDAAmD;;gBAEnD,4CAA4C;;gBAE5C,8DAA8D;;;;;;;;;;YAiBlE,yBAAyB;;gBAjFzB,0CAA0C;;oBAEtC,yBAAyB;;;gBAG7B,+DAA+D;;oBAE3D,iCAAiC;;oBAEjC,oCAAoC;;oBAEpC,iCAAiC;;;gBAGrC,iEAAiE;;oBAE7D,gDAAgD;;oBAEhD,mCAAmC;;oBAEnC,+BAA+B;;oBAE/B,iDAAiD;;oBAEjD,wCAAwC;;oBAExC,8DAA8D;;;gBAGlE,mEAAmE;;oBAE/D,gDAAgD;;oBAEhD,mCAAmC;;oBAEnC,sBAAsB;;oBAEtB,2BAA2B;;oBAE3B,6BAA6B;;;;;oBAE7B,iDAAiD;;oBAEjD,+DAA+D;;oBAE/D,8DAA8D;;;gBAGlE,oEAAoE;;oBAEhE,gDAAgD;;oBAEhD,mCAAmC;;oBAEnC,sBAAsB;;oBAEtB,6DAA6D;;oBAE7D,mCAAmC;;oBAEnC,mDAAmD;;oBAEnD,4CAA4C;;oBAE5C,8DAA8D;;;;YAmBlE,6CAA6C;;gBAxH7C,0CAA0C;;gBAE1C,2CAA2C;;oBAEvC,yEAAyE;;;gBAG7E,wDAAwD;;oBAEpD,wBAAwB;;oBAExB,iCAAiC;;;gBAGrC,+DAA+D;;oBAE3D,4CAA4C;;oBAE5C,sCAAsC;;oBAEtC,yDAAyD;;;;YAsG7D,oCAAoC;;YAEpC,oCAAoC;;YAEpC,4CAA4C;;YAE5C,+BAA+B;;YAE/B,kCAAkC;;YAElC,6CAA6C;;YAE7C,kDAAkD;;YAElD,sCAAsC;;YAEtC,0CAA0C;;;;YA4B1C,yBAAyB;;;oBAnBrB,mDAAmD;;oBAEnD,sBAAsB;;oBAEtB,uBAAuB;;oBAEvB,wBAAwB;;oBAExB,kBAAkB;;oBAElB,0CAA0C;;;;YAW9C,0BAA0B;;YAE1B,4BAA4B;;YAE5B,sCAAsC;;;;;gBAzBlC,mDAAmD;;gBAEnD,sBAAsB;;gBAEtB,uBAAuB;;gBAEvB,wBAAwB;;gBAExB,kBAAkB;;gBAElB,0CAA0C;;;;;;YA+B9C,uDAAuD;;gBAEnD,qEAAqE;;gBAErE,mFAAmF;;;YAGvF,0CAA0C;;;;YAoF1C,kCAAkC;;YAElC,gEAAgE;;YAEhE,+DAA+D;;;;CAmNzD,CAAC"}
|
|
@@ -0,0 +1,468 @@
|
|
|
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
|
+
* Lightning platform functions for East.
|
|
7
|
+
*
|
|
8
|
+
* Provides production-grade neural network training using PyTorch Lightning.
|
|
9
|
+
* Supports regression, binary classification, multiclass classification,
|
|
10
|
+
* and multi-head categorical outputs.
|
|
11
|
+
*
|
|
12
|
+
* @packageDocumentation
|
|
13
|
+
*/
|
|
14
|
+
import { East, StructType, VariantType, OptionType, IntegerType, FloatType, BlobType, ArrayType, NullType, BooleanType, FunctionType, StringType, } from "@elaraai/east";
|
|
15
|
+
import { VectorType, MatrixType } from "../types.js";
|
|
16
|
+
// Re-export shared types
|
|
17
|
+
export { VectorType, MatrixType } from "../types.js";
|
|
18
|
+
// ===========================================
|
|
19
|
+
// Type Definitions
|
|
20
|
+
// ===========================================
|
|
21
|
+
/**
|
|
22
|
+
* Lightning output mode - determines loss function and output activation.
|
|
23
|
+
*/
|
|
24
|
+
export const LightningOutputType = VariantType({
|
|
25
|
+
/** Regression: MSE loss, no activation */
|
|
26
|
+
regression: NullType,
|
|
27
|
+
/** Binary: BCE loss, sigmoid activation */
|
|
28
|
+
binary: StructType({
|
|
29
|
+
/** Optional per-position pos_weights for class imbalance [output_dim] */
|
|
30
|
+
pos_weight: OptionType(VectorType(FloatType)),
|
|
31
|
+
}),
|
|
32
|
+
/** Multiclass: CrossEntropy loss, softmax activation */
|
|
33
|
+
multiclass: StructType({
|
|
34
|
+
/** Number of classes */
|
|
35
|
+
n_classes: IntegerType,
|
|
36
|
+
/** Optional per-class weights */
|
|
37
|
+
class_weights: OptionType(VectorType(FloatType)),
|
|
38
|
+
}),
|
|
39
|
+
/** Multi-head categorical: N independent CrossEntropy heads */
|
|
40
|
+
multi_head: StructType({
|
|
41
|
+
/** Number of heads (e.g., 84 time slots) */
|
|
42
|
+
n_heads: IntegerType,
|
|
43
|
+
/** Classes per head (e.g., 4 bins) */
|
|
44
|
+
n_classes_per_head: IntegerType,
|
|
45
|
+
/** Optional class weights matrix (n_heads, n_classes) */
|
|
46
|
+
class_weights: OptionType(MatrixType(FloatType)),
|
|
47
|
+
}),
|
|
48
|
+
});
|
|
49
|
+
/**
|
|
50
|
+
* Cell type for sequential architectures.
|
|
51
|
+
*/
|
|
52
|
+
export const CellType = VariantType({
|
|
53
|
+
lstm: NullType,
|
|
54
|
+
gru: NullType,
|
|
55
|
+
});
|
|
56
|
+
/**
|
|
57
|
+
* Lightning architecture type.
|
|
58
|
+
*/
|
|
59
|
+
export const LightningArchitectureType = VariantType({
|
|
60
|
+
/** Simple MLP: input → hidden → output */
|
|
61
|
+
mlp: StructType({
|
|
62
|
+
/** Hidden layer sizes */
|
|
63
|
+
hidden_layers: ArrayType(IntegerType),
|
|
64
|
+
}),
|
|
65
|
+
/** Autoencoder: input → encoder → latent → decoder → output */
|
|
66
|
+
autoencoder: StructType({
|
|
67
|
+
/** Encoder hidden layer sizes */
|
|
68
|
+
encoder_layers: ArrayType(IntegerType),
|
|
69
|
+
/** Latent dimension (bottleneck) */
|
|
70
|
+
latent_dim: IntegerType,
|
|
71
|
+
/** Decoder hidden layer sizes */
|
|
72
|
+
decoder_layers: ArrayType(IntegerType),
|
|
73
|
+
}),
|
|
74
|
+
/** Conv1D: 1D convolutional autoencoder for temporal patterns */
|
|
75
|
+
conv1d: StructType({
|
|
76
|
+
/** Number of channels (e.g., additive types) */
|
|
77
|
+
n_channels: IntegerType,
|
|
78
|
+
/** Sequence length (e.g., days) */
|
|
79
|
+
sequence_length: IntegerType,
|
|
80
|
+
/** Conv layer channel sizes */
|
|
81
|
+
conv_channels: ArrayType(IntegerType),
|
|
82
|
+
/** Kernel size for convolutions (must be odd) */
|
|
83
|
+
kernel_size: IntegerType,
|
|
84
|
+
/** Latent dimension after flattening */
|
|
85
|
+
latent_dim: IntegerType,
|
|
86
|
+
/** Optional condition dimension for conditional generation */
|
|
87
|
+
condition_dim: OptionType(IntegerType),
|
|
88
|
+
}),
|
|
89
|
+
/** Sequential: LSTM/GRU autoencoder for long-range dependencies */
|
|
90
|
+
sequential: StructType({
|
|
91
|
+
/** Number of channels (e.g., additive types) */
|
|
92
|
+
n_channels: IntegerType,
|
|
93
|
+
/** Sequence length (e.g., days) */
|
|
94
|
+
sequence_length: IntegerType,
|
|
95
|
+
/** RNN hidden size */
|
|
96
|
+
hidden_size: IntegerType,
|
|
97
|
+
/** Number of RNN layers */
|
|
98
|
+
n_layers: IntegerType,
|
|
99
|
+
/** Cell type: lstm or gru */
|
|
100
|
+
cell_type: CellType,
|
|
101
|
+
/** Latent dimension (from final hidden state) */
|
|
102
|
+
latent_dim: IntegerType,
|
|
103
|
+
/** Bidirectional encoder (decoder is always unidirectional) */
|
|
104
|
+
bidirectional: BooleanType,
|
|
105
|
+
/** Optional condition dimension for conditional generation */
|
|
106
|
+
condition_dim: OptionType(IntegerType),
|
|
107
|
+
}),
|
|
108
|
+
/** Transformer: attention-based autoencoder for complex patterns */
|
|
109
|
+
transformer: StructType({
|
|
110
|
+
/** Number of channels (e.g., additive types) */
|
|
111
|
+
n_channels: IntegerType,
|
|
112
|
+
/** Sequence length (e.g., days) */
|
|
113
|
+
sequence_length: IntegerType,
|
|
114
|
+
/** Model dimension */
|
|
115
|
+
d_model: IntegerType,
|
|
116
|
+
/** Number of attention heads (must divide d_model evenly) */
|
|
117
|
+
n_attention_heads: IntegerType,
|
|
118
|
+
/** Number of transformer layers */
|
|
119
|
+
n_layers: IntegerType,
|
|
120
|
+
/** Feedforward dimension (default: 4 * d_model) */
|
|
121
|
+
d_ff: OptionType(IntegerType),
|
|
122
|
+
/** Latent dimension (mean pooled output) */
|
|
123
|
+
latent_dim: IntegerType,
|
|
124
|
+
/** Optional condition dimension for conditional generation */
|
|
125
|
+
condition_dim: OptionType(IntegerType),
|
|
126
|
+
}),
|
|
127
|
+
});
|
|
128
|
+
/**
|
|
129
|
+
* Epoch callback function type: (epoch, train_loss, val_loss) -> void
|
|
130
|
+
*/
|
|
131
|
+
export const LightningEpochCallbackType = FunctionType([IntegerType, FloatType, FloatType], NullType);
|
|
132
|
+
/**
|
|
133
|
+
* Lightning training configuration.
|
|
134
|
+
*/
|
|
135
|
+
export const LightningConfigType = StructType({
|
|
136
|
+
/** Model architecture */
|
|
137
|
+
architecture: LightningArchitectureType,
|
|
138
|
+
/** Output mode (determines loss function) */
|
|
139
|
+
output: LightningOutputType,
|
|
140
|
+
/** Learning rate (default: 1e-3) */
|
|
141
|
+
learning_rate: OptionType(FloatType),
|
|
142
|
+
/** Maximum epochs (default: 100) */
|
|
143
|
+
max_epochs: OptionType(IntegerType),
|
|
144
|
+
/** Early stopping patience (default: 10) */
|
|
145
|
+
patience: OptionType(IntegerType),
|
|
146
|
+
/** Batch size (default: 32) */
|
|
147
|
+
batch_size: OptionType(IntegerType),
|
|
148
|
+
/** Dropout rate (default: 0.1) */
|
|
149
|
+
dropout: OptionType(FloatType),
|
|
150
|
+
/** Gradient clipping value (default: 1.0) */
|
|
151
|
+
gradient_clip: OptionType(FloatType),
|
|
152
|
+
/** L2 regularization weight decay (default: 0) */
|
|
153
|
+
weight_decay: OptionType(FloatType),
|
|
154
|
+
/** Random seed for reproducibility */
|
|
155
|
+
random_state: OptionType(IntegerType),
|
|
156
|
+
/** Optional callback called each epoch */
|
|
157
|
+
epoch_callback: OptionType(LightningEpochCallbackType),
|
|
158
|
+
});
|
|
159
|
+
/**
|
|
160
|
+
* Lightning model blob structure.
|
|
161
|
+
*/
|
|
162
|
+
export const LightningModelBlobType = VariantType({
|
|
163
|
+
lightning: StructType({
|
|
164
|
+
/** Serialized model data (state_dict + hparams) */
|
|
165
|
+
data: BlobType,
|
|
166
|
+
/** Input dimension */
|
|
167
|
+
n_features: IntegerType,
|
|
168
|
+
/** Output dimension */
|
|
169
|
+
output_dim: IntegerType,
|
|
170
|
+
/** Architecture type */
|
|
171
|
+
architecture_type: StringType,
|
|
172
|
+
/** Output type */
|
|
173
|
+
output_type: StringType,
|
|
174
|
+
/** Latent dimension (autoencoder only) */
|
|
175
|
+
latent_dim: OptionType(IntegerType),
|
|
176
|
+
}),
|
|
177
|
+
});
|
|
178
|
+
/**
|
|
179
|
+
* Lightning training result.
|
|
180
|
+
*/
|
|
181
|
+
export const LightningResultType = StructType({
|
|
182
|
+
/** Trained model blob */
|
|
183
|
+
model: LightningModelBlobType,
|
|
184
|
+
/** Final training loss */
|
|
185
|
+
train_loss: FloatType,
|
|
186
|
+
/** Final validation loss */
|
|
187
|
+
val_loss: FloatType,
|
|
188
|
+
/** Best epoch (for early stopping) */
|
|
189
|
+
best_epoch: IntegerType,
|
|
190
|
+
});
|
|
191
|
+
/**
|
|
192
|
+
* 3D boolean tensor for masks: (n_samples, n_heads, n_classes)
|
|
193
|
+
*/
|
|
194
|
+
export const Tensor3DBoolType = ArrayType(ArrayType(ArrayType(BooleanType)));
|
|
195
|
+
/**
|
|
196
|
+
* Group-based weights for per-sample class weighting.
|
|
197
|
+
*
|
|
198
|
+
* Instead of per-sample weights (memory-intensive), samples belong to discrete
|
|
199
|
+
* groups (e.g., grades) with different weight configurations per group.
|
|
200
|
+
*/
|
|
201
|
+
export const GroupWeightsType = StructType({
|
|
202
|
+
/** Weights per group - shape depends on output type */
|
|
203
|
+
weights: VariantType({
|
|
204
|
+
/** For binary: pos_weight vector per group [n_groups][output_dim] */
|
|
205
|
+
binary: ArrayType(ArrayType(FloatType)),
|
|
206
|
+
/** For multi_head: class_weight matrix per group [n_groups][n_heads][n_classes] */
|
|
207
|
+
multi_head: ArrayType(ArrayType(ArrayType(FloatType))),
|
|
208
|
+
}),
|
|
209
|
+
/** Group index per sample: [n_samples] */
|
|
210
|
+
sample_groups: ArrayType(IntegerType),
|
|
211
|
+
});
|
|
212
|
+
// ===========================================
|
|
213
|
+
// Platform Functions
|
|
214
|
+
// ===========================================
|
|
215
|
+
/**
|
|
216
|
+
* Train a Lightning model.
|
|
217
|
+
*
|
|
218
|
+
* @param X - Input features matrix (n_samples, n_features)
|
|
219
|
+
* @param y - Target matrix (n_samples, output_dim)
|
|
220
|
+
* @param config - Training configuration
|
|
221
|
+
* @param masks - Optional 3D boolean masks (n_samples, n_heads, n_classes)
|
|
222
|
+
* @param group_weights - Optional group-based weights for per-sample weighting
|
|
223
|
+
* @param conditions - Optional condition matrix for conditional generation (n_samples, condition_dim)
|
|
224
|
+
* @returns Training result with model blob and metrics
|
|
225
|
+
*/
|
|
226
|
+
export const lightning_train = East.platform("lightning_train", [MatrixType(FloatType), MatrixType(FloatType), LightningConfigType, OptionType(Tensor3DBoolType), OptionType(GroupWeightsType), OptionType(MatrixType(FloatType))], LightningResultType);
|
|
227
|
+
/**
|
|
228
|
+
* Predict using a Lightning model.
|
|
229
|
+
*
|
|
230
|
+
* @param model - Trained model blob
|
|
231
|
+
* @param X - Input features matrix (n_samples, n_features)
|
|
232
|
+
* @param masks - Optional 3D boolean masks for inference
|
|
233
|
+
* @param conditions - Optional condition matrix for conditional models (n_samples, condition_dim)
|
|
234
|
+
* @returns Predicted probabilities matrix (n_samples, output_dim)
|
|
235
|
+
*/
|
|
236
|
+
export const lightning_predict = East.platform("lightning_predict", [LightningModelBlobType, MatrixType(FloatType), OptionType(Tensor3DBoolType), OptionType(MatrixType(FloatType))], MatrixType(FloatType));
|
|
237
|
+
/**
|
|
238
|
+
* Encode input to latent space (autoencoder only).
|
|
239
|
+
*
|
|
240
|
+
* @param model - Trained autoencoder model blob
|
|
241
|
+
* @param X - Input features matrix (n_samples, n_features)
|
|
242
|
+
* @returns Latent embeddings matrix (n_samples, latent_dim)
|
|
243
|
+
*/
|
|
244
|
+
export const lightning_encode = East.platform("lightning_encode", [LightningModelBlobType, MatrixType(FloatType)], MatrixType(FloatType));
|
|
245
|
+
/**
|
|
246
|
+
* Decode latent to output (autoencoder only).
|
|
247
|
+
*
|
|
248
|
+
* @param model - Trained autoencoder model blob
|
|
249
|
+
* @param z - Latent embeddings matrix (n_samples, latent_dim)
|
|
250
|
+
* @returns Decoded output matrix (n_samples, output_dim)
|
|
251
|
+
*/
|
|
252
|
+
export const lightning_decode = East.platform("lightning_decode", [LightningModelBlobType, MatrixType(FloatType)], MatrixType(FloatType));
|
|
253
|
+
/**
|
|
254
|
+
* Decode latent to output with condition (temporal architectures with condition_dim).
|
|
255
|
+
*
|
|
256
|
+
* @param model - Trained model with condition_dim set
|
|
257
|
+
* @param z - Latent embeddings matrix (n_samples, latent_dim)
|
|
258
|
+
* @param condition - Condition vectors (n_samples, condition_dim)
|
|
259
|
+
* @returns Decoded output matrix (n_samples, output_dim)
|
|
260
|
+
*/
|
|
261
|
+
export const lightning_decode_conditional = East.platform("lightning_decode_conditional", [LightningModelBlobType, MatrixType(FloatType), MatrixType(FloatType)], MatrixType(FloatType));
|
|
262
|
+
/**
|
|
263
|
+
* Configuration for autoregressive sequence generation.
|
|
264
|
+
*/
|
|
265
|
+
export const LightningGenerateConfigType = StructType({
|
|
266
|
+
/** Number of steps to generate */
|
|
267
|
+
n_steps: IntegerType,
|
|
268
|
+
/** Sampling temperature: 0.0 = argmax, > 0 = scaled sampling */
|
|
269
|
+
temperature: FloatType,
|
|
270
|
+
/** If true, return probabilities. If false, return samples. */
|
|
271
|
+
return_probs: BooleanType,
|
|
272
|
+
});
|
|
273
|
+
/**
|
|
274
|
+
* Generate sequence autoregressively from a sequential model.
|
|
275
|
+
*
|
|
276
|
+
* Shapes:
|
|
277
|
+
* - prefix: (n_prefix_steps, n_channels) - partial history to continue from, can be empty []
|
|
278
|
+
* - condition: (1, condition_dim) - conditioning features, or none
|
|
279
|
+
* - returns: (n_steps, n_channels) - generated timesteps only (not including prefix)
|
|
280
|
+
*
|
|
281
|
+
* @param model - Trained sequential model blob
|
|
282
|
+
* @param prefix - Partial history to continue from
|
|
283
|
+
* @param condition - Optional conditioning features
|
|
284
|
+
* @param config - Generation configuration
|
|
285
|
+
* @returns Generated sequence matrix
|
|
286
|
+
*/
|
|
287
|
+
export const lightning_generate_sequence = East.platform("lightning_generate_sequence", [LightningModelBlobType, MatrixType(FloatType), OptionType(MatrixType(FloatType)), LightningGenerateConfigType], MatrixType(FloatType));
|
|
288
|
+
// ===========================================
|
|
289
|
+
// Grouped Export
|
|
290
|
+
// ===========================================
|
|
291
|
+
/**
|
|
292
|
+
* Lightning types namespace.
|
|
293
|
+
*/
|
|
294
|
+
export const LightningTypes = {
|
|
295
|
+
OutputType: LightningOutputType,
|
|
296
|
+
ArchitectureType: LightningArchitectureType,
|
|
297
|
+
CellType,
|
|
298
|
+
EpochCallbackType: LightningEpochCallbackType,
|
|
299
|
+
ConfigType: LightningConfigType,
|
|
300
|
+
ResultType: LightningResultType,
|
|
301
|
+
ModelBlobType: LightningModelBlobType,
|
|
302
|
+
Tensor3DBoolType,
|
|
303
|
+
GroupWeightsType,
|
|
304
|
+
GenerateConfigType: LightningGenerateConfigType,
|
|
305
|
+
};
|
|
306
|
+
/**
|
|
307
|
+
* Lightning platform functions namespace.
|
|
308
|
+
*
|
|
309
|
+
* Provides production-grade neural network training using PyTorch Lightning.
|
|
310
|
+
*
|
|
311
|
+
* @example
|
|
312
|
+
* ```typescript
|
|
313
|
+
* const result = Lightning.train(X, y, {
|
|
314
|
+
* architecture: variant("autoencoder", {
|
|
315
|
+
* encoder_layers: [64n],
|
|
316
|
+
* latent_dim: 16n,
|
|
317
|
+
* decoder_layers: [64n],
|
|
318
|
+
* }),
|
|
319
|
+
* output: variant("multi_head", {
|
|
320
|
+
* n_heads: 84n,
|
|
321
|
+
* n_classes_per_head: 4n,
|
|
322
|
+
* class_weights: variant("none", null),
|
|
323
|
+
* }),
|
|
324
|
+
* }, variant("none", null));
|
|
325
|
+
*
|
|
326
|
+
* const embeddings = Lightning.encode(result.model, X);
|
|
327
|
+
* const predictions = Lightning.predict(result.model, X, variant("none", null));
|
|
328
|
+
* ```
|
|
329
|
+
*/
|
|
330
|
+
export const Lightning = {
|
|
331
|
+
/**
|
|
332
|
+
* Train a Lightning model.
|
|
333
|
+
*
|
|
334
|
+
* Trains a neural network using PyTorch Lightning with early stopping,
|
|
335
|
+
* gradient clipping, and optional epoch callbacks.
|
|
336
|
+
*
|
|
337
|
+
* @example
|
|
338
|
+
* ```ts
|
|
339
|
+
* import { East, FloatType, MatrixType, variant } from "@elaraai/east";
|
|
340
|
+
* import { Lightning, LightningConfigType } from "@elaraai/east-py-datascience";
|
|
341
|
+
*
|
|
342
|
+
* const train = East.function([], Lightning.Types.ResultType, ($) => {
|
|
343
|
+
* const X = $.let([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0], [7.0, 8.0]]);
|
|
344
|
+
* const y = $.let([[1.0, 0.0], [0.0, 1.0], [1.0, 0.0], [0.0, 1.0]]);
|
|
345
|
+
* const config = $.let({
|
|
346
|
+
* architecture: variant("autoencoder", {
|
|
347
|
+
* encoder_layers: new BigInt64Array([16n]),
|
|
348
|
+
* latent_dim: 4n,
|
|
349
|
+
* decoder_layers: new BigInt64Array([16n]),
|
|
350
|
+
* }),
|
|
351
|
+
* output: variant("regression", null),
|
|
352
|
+
* max_epochs: variant("some", 50n),
|
|
353
|
+
* batch_size: variant("some", 4n),
|
|
354
|
+
* learning_rate: variant("some", 0.001),
|
|
355
|
+
* patience: variant("some", 10n),
|
|
356
|
+
* seed: variant("some", 42n),
|
|
357
|
+
* epoch_callback: variant("none", null),
|
|
358
|
+
* gradient_clip_val: variant("none", null),
|
|
359
|
+
* condition_dim: variant("none", null),
|
|
360
|
+
* }, LightningConfigType);
|
|
361
|
+
* return $.return(Lightning.train(X, y, config, variant("none", null), variant("none", null), variant("none", null)));
|
|
362
|
+
* });
|
|
363
|
+
* ```
|
|
364
|
+
*/
|
|
365
|
+
train: lightning_train,
|
|
366
|
+
/**
|
|
367
|
+
* Predict using a trained Lightning model.
|
|
368
|
+
*
|
|
369
|
+
* Returns predictions with optional mask support for multi-head outputs.
|
|
370
|
+
*
|
|
371
|
+
* @example
|
|
372
|
+
* ```ts
|
|
373
|
+
* import { East, FloatType, MatrixType, variant } from "@elaraai/east";
|
|
374
|
+
* import { Lightning } from "@elaraai/east-py-datascience";
|
|
375
|
+
*
|
|
376
|
+
* const predictFn = East.function(
|
|
377
|
+
* [Lightning.Types.ModelBlobType, MatrixType(FloatType)],
|
|
378
|
+
* MatrixType(FloatType),
|
|
379
|
+
* ($, model, X) => {
|
|
380
|
+
* return $.return(Lightning.predict(model, X, variant("none", null), variant("none", null)));
|
|
381
|
+
* }
|
|
382
|
+
* );
|
|
383
|
+
* ```
|
|
384
|
+
*/
|
|
385
|
+
predict: lightning_predict,
|
|
386
|
+
/**
|
|
387
|
+
* Encode inputs to latent space (autoencoder only).
|
|
388
|
+
*
|
|
389
|
+
* @example
|
|
390
|
+
* ```ts
|
|
391
|
+
* import { East, FloatType, MatrixType } from "@elaraai/east";
|
|
392
|
+
* import { Lightning } from "@elaraai/east-py-datascience";
|
|
393
|
+
*
|
|
394
|
+
* const encodeFn = East.function(
|
|
395
|
+
* [Lightning.Types.ModelBlobType, MatrixType(FloatType)],
|
|
396
|
+
* MatrixType(FloatType),
|
|
397
|
+
* ($, model, X) => {
|
|
398
|
+
* // Returns (n_samples x latent_dim) embeddings
|
|
399
|
+
* return $.return(Lightning.encode(model, X));
|
|
400
|
+
* }
|
|
401
|
+
* );
|
|
402
|
+
* ```
|
|
403
|
+
*/
|
|
404
|
+
encode: lightning_encode,
|
|
405
|
+
/**
|
|
406
|
+
* Decode latent embeddings to output space (autoencoder only).
|
|
407
|
+
*
|
|
408
|
+
* @example
|
|
409
|
+
* ```ts
|
|
410
|
+
* import { East, FloatType, MatrixType } from "@elaraai/east";
|
|
411
|
+
* import { Lightning } from "@elaraai/east-py-datascience";
|
|
412
|
+
*
|
|
413
|
+
* const decodeFn = East.function(
|
|
414
|
+
* [Lightning.Types.ModelBlobType, MatrixType(FloatType)],
|
|
415
|
+
* MatrixType(FloatType),
|
|
416
|
+
* ($, model, z) => {
|
|
417
|
+
* // z is (n_samples x latent_dim), returns (n_samples x output_dim)
|
|
418
|
+
* return $.return(Lightning.decode(model, z));
|
|
419
|
+
* }
|
|
420
|
+
* );
|
|
421
|
+
* ```
|
|
422
|
+
*/
|
|
423
|
+
decode: lightning_decode,
|
|
424
|
+
/**
|
|
425
|
+
* Decode latent embeddings with condition vector (temporal architectures).
|
|
426
|
+
*
|
|
427
|
+
* @example
|
|
428
|
+
* ```ts
|
|
429
|
+
* import { East, FloatType, MatrixType } from "@elaraai/east";
|
|
430
|
+
* import { Lightning } from "@elaraai/east-py-datascience";
|
|
431
|
+
*
|
|
432
|
+
* const decodeFn = East.function(
|
|
433
|
+
* [Lightning.Types.ModelBlobType, MatrixType(FloatType), MatrixType(FloatType)],
|
|
434
|
+
* MatrixType(FloatType),
|
|
435
|
+
* ($, model, z, condition) => {
|
|
436
|
+
* // z: (n_samples, latent_dim), condition: (n_samples, condition_dim)
|
|
437
|
+
* return $.return(Lightning.decodeConditional(model, z, condition));
|
|
438
|
+
* }
|
|
439
|
+
* );
|
|
440
|
+
* ```
|
|
441
|
+
*/
|
|
442
|
+
decodeConditional: lightning_decode_conditional,
|
|
443
|
+
/**
|
|
444
|
+
* Generate sequence autoregressively from a sequential model.
|
|
445
|
+
*
|
|
446
|
+
* @example
|
|
447
|
+
* ```ts
|
|
448
|
+
* import { East, FloatType, IntegerType, BooleanType, MatrixType, variant } from "@elaraai/east";
|
|
449
|
+
* import { Lightning } from "@elaraai/east-py-datascience";
|
|
450
|
+
*
|
|
451
|
+
* const generateFn = East.function(
|
|
452
|
+
* [Lightning.Types.ModelBlobType, MatrixType(FloatType)],
|
|
453
|
+
* MatrixType(FloatType),
|
|
454
|
+
* ($, model, prefix) => {
|
|
455
|
+
* const config = $.let({ n_steps: 10n, temperature: 1.0, return_probs: false });
|
|
456
|
+
* // prefix: partial history, condition: none
|
|
457
|
+
* return $.return(Lightning.generateSequence(model, prefix, variant("none", null), config));
|
|
458
|
+
* }
|
|
459
|
+
* );
|
|
460
|
+
* ```
|
|
461
|
+
*/
|
|
462
|
+
generateSequence: lightning_generate_sequence,
|
|
463
|
+
/**
|
|
464
|
+
* Type definitions for Lightning functions.
|
|
465
|
+
*/
|
|
466
|
+
Types: LightningTypes,
|
|
467
|
+
};
|
|
468
|
+
//# sourceMappingURL=lightning.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"lightning.js","sourceRoot":"","sources":["../../../src/lightning/lightning.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;;;;;GAQG;AAEH,OAAO,EACH,IAAI,EACJ,UAAU,EACV,WAAW,EACX,UAAU,EACV,WAAW,EACX,SAAS,EACT,QAAQ,EACR,SAAS,EACT,QAAQ,EACR,WAAW,EACX,YAAY,EACZ,UAAU,GACb,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,yBAAyB;AACzB,OAAO,EAAE,UAAU,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAErD,8CAA8C;AAC9C,mBAAmB;AACnB,8CAA8C;AAE9C;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,WAAW,CAAC;IAC3C,0CAA0C;IAC1C,UAAU,EAAE,QAAQ;IACpB,2CAA2C;IAC3C,MAAM,EAAE,UAAU,CAAC;QACf,yEAAyE;QACzE,UAAU,EAAE,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;KAChD,CAAC;IACF,wDAAwD;IACxD,UAAU,EAAE,UAAU,CAAC;QACnB,wBAAwB;QACxB,SAAS,EAAE,WAAW;QACtB,iCAAiC;QACjC,aAAa,EAAE,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;KACnD,CAAC;IACF,+DAA+D;IAC/D,UAAU,EAAE,UAAU,CAAC;QACnB,4CAA4C;QAC5C,OAAO,EAAE,WAAW;QACpB,sCAAsC;QACtC,kBAAkB,EAAE,WAAW;QAC/B,yDAAyD;QACzD,aAAa,EAAE,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;KACnD,CAAC;CACL,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG,WAAW,CAAC;IAChC,IAAI,EAAE,QAAQ;IACd,GAAG,EAAE,QAAQ;CAChB,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,WAAW,CAAC;IACjD,0CAA0C;IAC1C,GAAG,EAAE,UAAU,CAAC;QACZ,yBAAyB;QACzB,aAAa,EAAE,SAAS,CAAC,WAAW,CAAC;KACxC,CAAC;IACF,+DAA+D;IAC/D,WAAW,EAAE,UAAU,CAAC;QACpB,iCAAiC;QACjC,cAAc,EAAE,SAAS,CAAC,WAAW,CAAC;QACtC,oCAAoC;QACpC,UAAU,EAAE,WAAW;QACvB,iCAAiC;QACjC,cAAc,EAAE,SAAS,CAAC,WAAW,CAAC;KACzC,CAAC;IACF,iEAAiE;IACjE,MAAM,EAAE,UAAU,CAAC;QACf,gDAAgD;QAChD,UAAU,EAAE,WAAW;QACvB,mCAAmC;QACnC,eAAe,EAAE,WAAW;QAC5B,+BAA+B;QAC/B,aAAa,EAAE,SAAS,CAAC,WAAW,CAAC;QACrC,iDAAiD;QACjD,WAAW,EAAE,WAAW;QACxB,wCAAwC;QACxC,UAAU,EAAE,WAAW;QACvB,8DAA8D;QAC9D,aAAa,EAAE,UAAU,CAAC,WAAW,CAAC;KACzC,CAAC;IACF,mEAAmE;IACnE,UAAU,EAAE,UAAU,CAAC;QACnB,gDAAgD;QAChD,UAAU,EAAE,WAAW;QACvB,mCAAmC;QACnC,eAAe,EAAE,WAAW;QAC5B,sBAAsB;QACtB,WAAW,EAAE,WAAW;QACxB,2BAA2B;QAC3B,QAAQ,EAAE,WAAW;QACrB,6BAA6B;QAC7B,SAAS,EAAE,QAAQ;QACnB,iDAAiD;QACjD,UAAU,EAAE,WAAW;QACvB,+DAA+D;QAC/D,aAAa,EAAE,WAAW;QAC1B,8DAA8D;QAC9D,aAAa,EAAE,UAAU,CAAC,WAAW,CAAC;KACzC,CAAC;IACF,oEAAoE;IACpE,WAAW,EAAE,UAAU,CAAC;QACpB,gDAAgD;QAChD,UAAU,EAAE,WAAW;QACvB,mCAAmC;QACnC,eAAe,EAAE,WAAW;QAC5B,sBAAsB;QACtB,OAAO,EAAE,WAAW;QACpB,6DAA6D;QAC7D,iBAAiB,EAAE,WAAW;QAC9B,mCAAmC;QACnC,QAAQ,EAAE,WAAW;QACrB,mDAAmD;QACnD,IAAI,EAAE,UAAU,CAAC,WAAW,CAAC;QAC7B,4CAA4C;QAC5C,UAAU,EAAE,WAAW;QACvB,8DAA8D;QAC9D,aAAa,EAAE,UAAU,CAAC,WAAW,CAAC;KACzC,CAAC;CACL,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAG,YAAY,CAClD,CAAC,WAAW,EAAE,SAAS,EAAE,SAAS,CAAC,EACnC,QAAQ,CACX,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,UAAU,CAAC;IAC1C,yBAAyB;IACzB,YAAY,EAAE,yBAAyB;IACvC,6CAA6C;IAC7C,MAAM,EAAE,mBAAmB;IAC3B,oCAAoC;IACpC,aAAa,EAAE,UAAU,CAAC,SAAS,CAAC;IACpC,oCAAoC;IACpC,UAAU,EAAE,UAAU,CAAC,WAAW,CAAC;IACnC,4CAA4C;IAC5C,QAAQ,EAAE,UAAU,CAAC,WAAW,CAAC;IACjC,+BAA+B;IAC/B,UAAU,EAAE,UAAU,CAAC,WAAW,CAAC;IACnC,kCAAkC;IAClC,OAAO,EAAE,UAAU,CAAC,SAAS,CAAC;IAC9B,6CAA6C;IAC7C,aAAa,EAAE,UAAU,CAAC,SAAS,CAAC;IACpC,kDAAkD;IAClD,YAAY,EAAE,UAAU,CAAC,SAAS,CAAC;IACnC,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,0CAA0C;IAC1C,cAAc,EAAE,UAAU,CAAC,0BAA0B,CAAC;CACzD,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG,WAAW,CAAC;IAC9C,SAAS,EAAE,UAAU,CAAC;QAClB,mDAAmD;QACnD,IAAI,EAAE,QAAQ;QACd,sBAAsB;QACtB,UAAU,EAAE,WAAW;QACvB,uBAAuB;QACvB,UAAU,EAAE,WAAW;QACvB,wBAAwB;QACxB,iBAAiB,EAAE,UAAU;QAC7B,kBAAkB;QAClB,WAAW,EAAE,UAAU;QACvB,0CAA0C;QAC1C,UAAU,EAAE,UAAU,CAAC,WAAW,CAAC;KACtC,CAAC;CACL,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,mBAAmB,GAAG,UAAU,CAAC;IAC1C,yBAAyB;IACzB,KAAK,EAAE,sBAAsB;IAC7B,0BAA0B;IAC1B,UAAU,EAAE,SAAS;IACrB,4BAA4B;IAC5B,QAAQ,EAAE,SAAS;IACnB,sCAAsC;IACtC,UAAU,EAAE,WAAW;CAC1B,CAAC,CAAC;AAEH;;GAEG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;AAE7E;;;;;GAKG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,UAAU,CAAC;IACvC,uDAAuD;IACvD,OAAO,EAAE,WAAW,CAAC;QACjB,qEAAqE;QACrE,MAAM,EAAE,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QACvC,mFAAmF;QACnF,UAAU,EAAE,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,CAAC;KACzD,CAAC;IACF,0CAA0C;IAC1C,aAAa,EAAE,SAAS,CAAC,WAAW,CAAC;CACxC,CAAC,CAAC;AAEH,8CAA8C;AAC9C,qBAAqB;AACrB,8CAA8C;AAE9C;;;;;;;;;;GAUG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,IAAI,CAAC,QAAQ,CACxC,iBAAiB,EACjB,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,mBAAmB,EAAE,UAAU,CAAC,gBAAgB,CAAC,EAAE,UAAU,CAAC,gBAAgB,CAAC,EAAE,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,EAClK,mBAAmB,CACtB,CAAC;AAEF;;;;;;;;GAQG;AACH,MAAM,CAAC,MAAM,iBAAiB,GAAG,IAAI,CAAC,QAAQ,CAC1C,mBAAmB,EACnB,CAAC,sBAAsB,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,gBAAgB,CAAC,EAAE,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,EAChH,UAAU,CAAC,SAAS,CAAC,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CACzC,kBAAkB,EAClB,CAAC,sBAAsB,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,EAC/C,UAAU,CAAC,SAAS,CAAC,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CACzC,kBAAkB,EAClB,CAAC,sBAAsB,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,EAC/C,UAAU,CAAC,SAAS,CAAC,CACxB,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,4BAA4B,GAAG,IAAI,CAAC,QAAQ,CACrD,8BAA8B,EAC9B,CAAC,sBAAsB,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,EACtE,UAAU,CAAC,SAAS,CAAC,CACxB,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,2BAA2B,GAAG,UAAU,CAAC;IAClD,kCAAkC;IAClC,OAAO,EAAE,WAAW;IACpB,gEAAgE;IAChE,WAAW,EAAE,SAAS;IACtB,+DAA+D;IAC/D,YAAY,EAAE,WAAW;CAC5B,CAAC,CAAC;AAEH;;;;;;;;;;;;;GAaG;AACH,MAAM,CAAC,MAAM,2BAA2B,GAAG,IAAI,CAAC,QAAQ,CACpD,6BAA6B,EAC7B,CAAC,sBAAsB,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,EAAE,2BAA2B,CAAC,EAC/G,UAAU,CAAC,SAAS,CAAC,CACxB,CAAC;AAEF,8CAA8C;AAC9C,iBAAiB;AACjB,8CAA8C;AAE9C;;GAEG;AACH,MAAM,CAAC,MAAM,cAAc,GAAG;IAC1B,UAAU,EAAE,mBAAmB;IAC/B,gBAAgB,EAAE,yBAAyB;IAC3C,QAAQ;IACR,iBAAiB,EAAE,0BAA0B;IAC7C,UAAU,EAAE,mBAAmB;IAC/B,UAAU,EAAE,mBAAmB;IAC/B,aAAa,EAAE,sBAAsB;IACrC,gBAAgB;IAChB,gBAAgB;IAChB,kBAAkB,EAAE,2BAA2B;CACzC,CAAC;AAEX;;;;;;;;;;;;;;;;;;;;;;;GAuBG;AACH,MAAM,CAAC,MAAM,SAAS,GAAG;IACrB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;OAiCG;IACH,KAAK,EAAE,eAAe;IAEtB;;;;;;;;;;;;;;;;;;OAkBG;IACH,OAAO,EAAE,iBAAiB;IAE1B;;;;;;;;;;;;;;;;;OAiBG;IACH,MAAM,EAAE,gBAAgB;IAExB;;;;;;;;;;;;;;;;;OAiBG;IACH,MAAM,EAAE,gBAAgB;IAExB;;;;;;;;;;;;;;;;;OAiBG;IACH,iBAAiB,EAAE,4BAA4B;IAE/C;;;;;;;;;;;;;;;;;;OAkBG;IACH,gBAAgB,EAAE,2BAA2B;IAE7C;;OAEG;IACH,KAAK,EAAE,cAAc;CACf,CAAC"}
|