@elaraai/east-py-datascience 0.0.2-beta.8 → 0.0.2-beta.80

package/dist/{lightgbm → src/lightgbm}/lightgbm.js

@@ -11,9 +11,9 @@
  * @packageDocumentation
  */
 import { East, StructType, VariantType, OptionType, IntegerType, FloatType, BlobType, } from "@elaraai/east";
-import { VectorType, MatrixType, LabelVectorType } from "../types.js";
+import { VectorType, MatrixType } from "../types.js";
 // Re-export shared types for convenience
-export { VectorType, MatrixType, LabelVectorType } from "../types.js";
+export { VectorType, MatrixType } from "../types.js";
 // ============================================================================
 // Config Types
 // ============================================================================
@@ -81,7 +81,7 @@ export const LightGBMModelBlobType = VariantType({
  * @param config - LightGBM configuration
  * @returns Model blob containing trained regressor
  */
-export const lightgbm_train_regressor = East.platform("lightgbm_train_regressor", [MatrixType, VectorType, LightGBMConfigType], LightGBMModelBlobType);
+export const lightgbm_train_regressor = East.platform("lightgbm_train_regressor", [MatrixType(FloatType), VectorType(FloatType), LightGBMConfigType], LightGBMModelBlobType);
 /**
  * Train a LightGBM classification model.
  *
@@ -90,7 +90,7 @@ export const lightgbm_train_regressor = East.platform("lightgbm_train_regressor"
  * @param config - LightGBM configuration
  * @returns Model blob containing trained classifier
  */
-export const lightgbm_train_classifier = East.platform("lightgbm_train_classifier", [MatrixType, LabelVectorType, LightGBMConfigType], LightGBMModelBlobType);
+export const lightgbm_train_classifier = East.platform("lightgbm_train_classifier", [MatrixType(FloatType), VectorType(IntegerType), LightGBMConfigType], LightGBMModelBlobType);
 /**
  * Make predictions with a trained LightGBM regressor.
  *
@@ -98,7 +98,7 @@ export const lightgbm_train_classifier = East.platform("lightgbm_train_classifie
  * @param X - Feature matrix
  * @returns Predicted values
  */
-export const lightgbm_predict = East.platform("lightgbm_predict", [LightGBMModelBlobType, MatrixType], VectorType);
+export const lightgbm_predict = East.platform("lightgbm_predict", [LightGBMModelBlobType, MatrixType(FloatType)], VectorType(FloatType));
 /**
  * Predict class labels with a trained LightGBM classifier.
  *
@@ -106,7 +106,7 @@ export const lightgbm_predict = East.platform("lightgbm_predict", [LightGBMModel
  * @param X - Feature matrix
  * @returns Predicted class labels
  */
-export const lightgbm_predict_class = East.platform("lightgbm_predict_class", [LightGBMModelBlobType, MatrixType], LabelVectorType);
+export const lightgbm_predict_class = East.platform("lightgbm_predict_class", [LightGBMModelBlobType, MatrixType(FloatType)], VectorType(IntegerType));
 /**
  * Get class probabilities from a trained LightGBM classifier.
  *
@@ -114,7 +114,7 @@ export const lightgbm_predict_class = East.platform("lightgbm_predict_class", [L
  * @param X - Feature matrix
  * @returns Probability matrix (n_samples x n_classes)
  */
-export const lightgbm_predict_proba = East.platform("lightgbm_predict_proba", [LightGBMModelBlobType, MatrixType], MatrixType);
+export const lightgbm_predict_proba = East.platform("lightgbm_predict_proba", [LightGBMModelBlobType, MatrixType(FloatType)], MatrixType(FloatType));
 // ============================================================================
 // Grouped Export
 // ============================================================================
@@ -122,12 +122,6 @@ export const lightgbm_predict_proba = East.platform("lightgbm_predict_proba", [L
  * Type definitions for LightGBM functions.
  */
 export const LightGBMTypes = {
-    /** Vector type (array of floats) */
-    VectorType,
-    /** Matrix type (2D array of floats) */
-    MatrixType,
-    /** Label vector type (array of integers) */
-    LabelVectorType,
     /** LightGBM configuration type */
     LightGBMConfigType,
     /** Model blob type for LightGBM models */
@@ -139,15 +133,107 @@ export const LightGBMTypes = {
  * Provides fast regression and classification with gradient boosted decision trees.
  */
 export const LightGBM = {
-    /** Train LightGBM regressor */
+    /**
+     * Train a LightGBM regression model.
+     *
+     * @example
+     * ```ts
+     * import { East, FloatType, MatrixType, VectorType, variant } from "@elaraai/east";
+     * import { LightGBM, LightGBMConfigType } from "@elaraai/east-py-datascience";
+     *
+     * const train = East.function([], LightGBM.Types.ModelBlobType, ($) => {
+     *     const X = $.let([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0], [7.0, 8.0]]);
+     *     const y = $.let(new Float64Array([1.0, 2.0, 3.0, 4.0]));
+     *     const config = $.let({
+     *         n_estimators: variant("some", 100n), max_depth: variant("some", 3n),
+     *         learning_rate: variant("some", 0.1), num_leaves: variant("none", null),
+     *         min_child_samples: variant("none", null), subsample: variant("none", null),
+     *         colsample_bytree: variant("none", null), reg_alpha: variant("none", null),
+     *         reg_lambda: variant("none", null), random_state: variant("some", 42n),
+     *         n_jobs: variant("none", null),
+     *     }, LightGBMConfigType);
+     *     return $.return(LightGBM.trainRegressor(X, y, config));
+     * });
+     * ```
+     */
     trainRegressor: lightgbm_train_regressor,
-    /** Train LightGBM classifier */
+    /**
+     * Train a LightGBM classification model.
+     *
+     * @example
+     * ```ts
+     * import { East, FloatType, IntegerType, MatrixType, VectorType, variant } from "@elaraai/east";
+     * import { LightGBM, LightGBMConfigType } from "@elaraai/east-py-datascience";
+     *
+     * const train = East.function([], LightGBM.Types.ModelBlobType, ($) => {
+     *     const X = $.let([[1.0, 2.0], [3.0, 4.0], [5.0, 6.0], [7.0, 8.0]]);
+     *     const y = $.let(new BigInt64Array([0n, 1n, 0n, 1n]));
+     *     const config = $.let({
+     *         n_estimators: variant("some", 50n), max_depth: variant("some", 3n),
+     *         learning_rate: variant("some", 0.1), num_leaves: variant("none", null),
+     *         min_child_samples: variant("none", null), subsample: variant("none", null),
+     *         colsample_bytree: variant("none", null), reg_alpha: variant("none", null),
+     *         reg_lambda: variant("none", null), random_state: variant("some", 42n),
+     *         n_jobs: variant("none", null),
+     *     }, LightGBMConfigType);
+     *     return $.return(LightGBM.trainClassifier(X, y, config));
+     * });
+     * ```
+     */
     trainClassifier: lightgbm_train_classifier,
-    /** Make predictions with regressor */
+    /**
+     * Make predictions with a trained LightGBM regressor.
+     *
+     * @example
+     * ```ts
+     * import { East, FloatType, MatrixType, VectorType } from "@elaraai/east";
+     * import { LightGBM } from "@elaraai/east-py-datascience";
+     *
+     * const predictFn = East.function(
+     *     [LightGBM.Types.ModelBlobType, MatrixType(FloatType)],
+     *     VectorType(FloatType),
+     *     ($, model, X) => {
+     *         return $.return(LightGBM.predict(model, X));
+     *     }
+     * );
+     * ```
+     */
     predict: lightgbm_predict,
-    /** Predict class labels with classifier */
+    /**
+     * Predict class labels with a trained LightGBM classifier.
+     *
+     * @example
+     * ```ts
+     * import { East, FloatType, IntegerType, MatrixType, VectorType } from "@elaraai/east";
+     * import { LightGBM } from "@elaraai/east-py-datascience";
+     *
+     * const predictFn = East.function(
+     *     [LightGBM.Types.ModelBlobType, MatrixType(FloatType)],
+     *     VectorType(IntegerType),
+     *     ($, model, X) => {
+     *         return $.return(LightGBM.predictClass(model, X));
+     *     }
+     * );
+     * ```
+     */
     predictClass: lightgbm_predict_class,
-    /** Get class probabilities from classifier */
+    /**
+     * Get class probabilities from a trained LightGBM classifier.
+     *
+     * @example
+     * ```ts
+     * import { East, FloatType, MatrixType } from "@elaraai/east";
+     * import { LightGBM } from "@elaraai/east-py-datascience";
+     *
+     * const predictFn = East.function(
+     *     [LightGBM.Types.ModelBlobType, MatrixType(FloatType)],
+     *     MatrixType(FloatType),
+     *     ($, model, X) => {
+     *         return $.return(LightGBM.predictProba(model, X));
+     *     }
+     * );
+     * ```
+     */
     predictProba: lightgbm_predict_proba,
     /** Type definitions */
     Types: LightGBMTypes,

package/dist/src/lightgbm/lightgbm.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"lightgbm.js","sourceRoot":"","sources":["../../../src/lightgbm/lightgbm.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,kBAAkB,GAAG,UAAU,CAAC;IACzC,8CAA8C;IAC9C,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,wDAAwD;IACxD,SAAS,EAAE,UAAU,CAAC,WAAW,CAAC;IAClC,wDAAwD;IACxD,aAAa,EAAE,UAAU,CAAC,SAAS,CAAC;IACpC,wDAAwD;IACxD,UAAU,EAAE,UAAU,CAAC,WAAW,CAAC;IACnC,gEAAgE;IAChE,iBAAiB,EAAE,UAAU,CAAC,WAAW,CAAC;IAC1C,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,sCAAsC;IACtC,YAAY,EAAE,UAAU,CAAC,WAAW,CAAC;IACrC,4DAA4D;IAC5D,MAAM,EAAE,UAAU,CAAC,WAAW,CAAC;CAClC,CAAC,CAAC;AAEH,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;GAIG;AACH,MAAM,CAAC,MAAM,qBAAqB,GAAG,WAAW,CAAC;IAC7C,+BAA+B;IAC/B,kBAAkB,EAAE,UAAU,CAAC;QAC3B,mCAAmC;QACnC,IAAI,EAAE,QAAQ;QACd,+BAA+B;QAC/B,UAAU,EAAE,WAAW;KAC1B,CAAC;IACF,gCAAgC;IAChC,mBAAmB,EAAE,UAAU,CAAC;QAC5B,mCAAmC;QACnC,IAAI,EAAE,QAAQ;QACd,+BAA+B;QAC/B,UAAU,EAAE,WAAW;QACvB,wBAAwB;QACxB,SAAS,EAAE,WAAW;KACzB,CAAC;CACL,CAAC,CAAC;AAEH,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAE/E;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,wBAAwB,GAAG,IAAI,CAAC,QAAQ,CACjD,0BAA0B,EAC1B,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,SAAS,CAAC,EAAE,kBAAkB,CAAC,EAClE,qBAAqB,CACxB,CAAC;AAEF;;;;;;;GAOG;AACH,MAAM,CAAC,MAAM,yBAAyB,GAAG,IAAI,CAAC,QAAQ,CAClD,2BAA2B,EAC3B,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,UAAU,CAAC,WAAW,CAAC,EAAE,kBAAkB,CAAC,EACpE,qBAAqB,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CACzC,kBAAkB,EAClB,CAAC,qBAAqB,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,EAC9C,UAAU,CAAC,SAAS,CAAC,CACxB,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG,IAAI,CAAC,QAAQ,CAC/C,wBAAwB,EACxB,CAAC,qBAAqB,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,EAC9C,UAAU,CAAC,WAAW,CAAC,CAC1B,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,sBAAsB,GAAG,IAAI,CAAC,QAAQ,CAC/C,wBAAwB,EACxB,CAAC,qBAAqB,EAAE,UAAU,CAAC,SAAS,CAAC,CAAC,EAC9C,UAAU,CAAC,SAAS,CAAC,CACxB,CAAC;AAEF,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG;IACzB,kCAAkC;IAClC,kBAAkB;IAClB,0CAA0C;IAC1C,aAAa,EAAE,qBAAqB;CAC9B,CAAC;AAEX;;;;GAIG;AACH,MAAM,CAAC,MAAM,QAAQ,GAAG;IACpB;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACH,cAAc,EAAE,wBAAwB;IAExC;;;;;;;;;;;;;;;;;;;;;;OAsBG;IACH,eAAe,EAAE,yBAAyB;IAE1C;;;;;;;;;;;;;;;;OAgBG;IACH,OAAO,EAAE,gBAAgB;IAEzB;;;;;;;;;;;;;;;;OAgBG;IACH,YAAY,EAAE,sBAAsB;IAEpC;;;;;;;;;;;;;;;;OAgBG;IACH,YAAY,EAAE,sBAAsB;IACpC,uBAAuB;IACvB,KAAK,EAAE,aAAa;CACd,CAAC"}