@wlearn/lightgbm 0.1.0 → 0.2.0

package/package.json

@@ -1,15 +1,19 @@
 {
   "name": "@wlearn/lightgbm",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "LightGBM WASM port for wlearn",
-  "type": "module",
+  "type": "commonjs",
   "main": "src/index.js",
   "exports": {
-    ".": "./src/index.js"
+    ".": {
+      "require": "./src/index.js",
+      "default": "./src/index.js"
+    }
   },
   "files": [
     "src/",
     "wasm/",
+    "dist/",
     "LICENSE",
     "README.md",
     "CHANGELOG.md"
@@ -19,17 +23,36 @@
     "access": "public"
   },
   "scripts": {
-    "test": "node --experimental-vm-modules test/test.js",
+    "test": "node test/test.js",
     "build": "bash scripts/build-wasm.sh",
-    "verify": "bash scripts/verify-exports.sh"
+    "verify": "bash scripts/verify-exports.sh",
+    "build:browser": "bash scripts/build-browser.sh",
+    "prepack": "node scripts/prepack.js",
+    "prepublishOnly": "node scripts/prepublish.js"
   },
   "dependencies": {
-    "@wlearn/types": "0.1.0",
-    "@wlearn/core": "0.1.0"
+    "@wlearn/core": "0.2.0"
   },
+  "keywords": [
+    "lightgbm",
+    "gradient-boosting",
+    "machine-learning",
+    "wasm",
+    "webassembly",
+    "wlearn"
+  ],
+  "author": "Anton Zemlyansky",
   "license": "MIT",
   "repository": {
     "type": "git",
     "url": "https://github.com/wlearn-org/lightgbm-wasm.git"
-  }
+  },
+  "devDependencies": {
+    "esbuild": "^0.27.3",
+    "playwright": "^1.58.2"
+  },
+  "bugs": {
+    "url": "https://github.com/wlearn-org/lightgbm-wasm/issues"
+  },
+  "homepage": "https://wlearn.org"
 }

package/src/booster.js

@@ -1,4 +1,4 @@
-import { getWasm } from './wasm.js'
+const { getWasm } = require('./wasm.js')
 
 // FinalizationRegistry safety net
 const registry = typeof FinalizationRegistry !== 'undefined'
@@ -28,7 +28,7 @@ function getLastError(wasm) {
 // Internal sentinel for loadModel path
 const LOAD_SENTINEL = Symbol('load')
 
-export class Booster {
+class Booster {
   #handle = null
   #freed = false
   #ptrRef = null
@@ -227,3 +227,5 @@ export class Booster {
     this.#handle = null
   }
 }
+
+module.exports = { Booster }

package/src/dataset.js

@@ -1,4 +1,4 @@
-import { getWasm } from './wasm.js'
+const { getWasm } = require('./wasm.js')
 
 // FinalizationRegistry safety net
 const registry = typeof FinalizationRegistry !== 'undefined'
@@ -25,7 +25,7 @@ function getLastError(wasm) {
   return wasm.UTF8ToString(wasm._wl_lgb_get_last_error())
 }
 
-export class Dataset {
+class Dataset {
   #handle = null
   #freed = false
   #ptrRef = null
@@ -101,3 +101,5 @@ export class Dataset {
     this.#handle = null
   }
 }
+
+module.exports = { Dataset }

package/src/index.js

@@ -1,4 +1,9 @@
-export { LGBModel } from './model.js'
-export { loadLGB } from './wasm.js'
-export { Dataset } from './dataset.js'
-export { Booster } from './booster.js'
+const { LGBModel: LGBModelImpl } = require('./model.js')
+const { loadLGB } = require('./wasm.js')
+const { Dataset } = require('./dataset.js')
+const { Booster } = require('./booster.js')
+const { createModelClass } = require('@wlearn/core')
+
+const LGBModel = createModelClass(LGBModelImpl, LGBModelImpl, { name: 'LGBModel', load: loadLGB })
+
+module.exports = { LGBModel, loadLGB, Dataset, Booster }

package/src/model.js

@@ -1,12 +1,12 @@
-import { loadLGB, getWasm } from './wasm.js'
-import { Dataset } from './dataset.js'
-import { Booster } from './booster.js'
-import {
+const { loadLGB, getWasm } = require('./wasm.js')
+const { Dataset } = require('./dataset.js')
+const { Booster } = require('./booster.js')
+const {
   normalizeY,
   encodeBundle, decodeBundle,
   register,
   DisposedError, NotFittedError
-} from '@wlearn/core'
+} = require('@wlearn/core')
 
 // FinalizationRegistry safety net
 const leakRegistry = typeof FinalizationRegistry !== 'undefined'
@@ -29,14 +29,14 @@ const PROBA_OBJECTIVES = new Set([
 ])
 
 // LightGBM params that are wlearn-only (not passed to Booster)
-const WLEARN_PARAMS = new Set(['numRound', 'coerce'])
+const WLEARN_PARAMS = new Set(['numRound', 'coerce', 'task'])
 
 // --- Internal sentinel for load path ---
 const LOAD_SENTINEL = Symbol('load')
 
 // --- LGBModel ---
 
-export class LGBModel {
+class LGBModel {
   #booster = null
   #freed = false
   #boosterRef = null
@@ -78,6 +78,9 @@ export class LGBModel {
   fit(X, y) {
     this.#ensureNotDisposed()
 
+    // Map task param to objective if needed
+    this.#resolveTask(y)
+
     // Dispose previous booster if refitting
     if (this.#booster) {
       this.#booster.dispose()
@@ -412,6 +415,29 @@ export class LGBModel {
     if (!this.#fitted) throw new NotFittedError('LGBModel is not fitted. Call fit() first.')
   }
 
+  #resolveTask(y) {
+    const task = this.#params.task
+    if (!task) return
+    // If objective is already set, it takes precedence over task
+    if (this.#params.objective) return
+    if (task === 'classification') {
+      // Count unique values in y to decide binary vs multiclass
+      const yNorm = normalizeY(y)
+      const unique = new Set()
+      for (let i = 0; i < yNorm.length; i++) unique.add(yNorm[i])
+      if (unique.size > 2) {
+        this.#params.objective = 'multiclass'
+        this.#params.num_class = unique.size
+      } else {
+        this.#params.objective = 'binary'
+      }
+    } else if (task === 'regression') {
+      this.#params.objective = 'regression'
+    } else {
+      throw new Error(`Unknown task: '${task}'. Use 'classification' or 'regression'.`)
+    }
+  }
+
   #isClassifier() {
     const obj = this.#params.objective || 'regression'
     return CLASSIFIER_OBJECTIVES.has(obj)
@@ -422,3 +448,5 @@ export class LGBModel {
 
 register('wlearn.lightgbm.classifier@1', async (m, t, b) => LGBModel._fromBundle(m, t, b))
 register('wlearn.lightgbm.regressor@1', async (m, t, b) => LGBModel._fromBundle(m, t, b))
+
+module.exports = { LGBModel }

package/src/wasm.js

@@ -1,19 +1,14 @@
 // WASM loader -- loads the LightGBM WASM module (singleton, lazy init)
 
-import { createRequire } from 'module'
-
 let wasmModule = null
 let loading = null
 
-export async function loadLGB(options = {}) {
+async function loadLGB(options = {}) {
   if (wasmModule) return wasmModule
   if (loading) return loading
 
   loading = (async () => {
-    // SINGLE_FILE=1: .wasm is embedded in the .cjs file, no locateFile needed
-    // Emscripten output is CJS, use createRequire for ESM compatibility
-    const require = createRequire(import.meta.url)
-    const createLightGBM = require('../wasm/lightgbm.cjs')
+    const createLightGBM = require('../wasm/lightgbm.js')
     wasmModule = await createLightGBM(options)
     return wasmModule
   })()
@@ -21,7 +16,9 @@ export async function loadLGB(options = {}) {
   return loading
 }
 
-export function getWasm() {
+function getWasm() {
   if (!wasmModule) throw new Error('WASM not loaded -- call loadLGB() first')
   return wasmModule
 }
+
+module.exports = { loadLGB, getWasm }

package/wasm/BUILD_INFO

@@ -1,6 +1,6 @@
 upstream: LightGBM v4.6.0
 upstream_commit: d02a01ac6f51d36c9e62388243bcb75c3b1b1774
-build_date: 2026-02-27T13:05:21Z
+build_date: 2026-05-25T14:51:03Z
 emscripten: emcc (Emscripten gcc/clang-like replacement + linker emulating GNU ld) 5.0.2 (dc80f645ee70178c11666de0c3860d9e064d50e4)
 build_flags: -O2 -fexceptions SINGLE_FILE=1
 wasm_embedded: true