An open API service indexing awesome lists of open source software.

https://github.com/fractal-solutions/xgboost-js

A pure JavaScript implementation of XGBoost for both Node.js and browser environments.
https://github.com/fractal-solutions/xgboost-js

ai artificial-intelligence bun bun-js deno javascript machine-learning nodejs npm supervised-learning xgboost xgboost-model

Last synced: 9 months ago
JSON representation

A pure JavaScript implementation of XGBoost for both Node.js and browser environments.

Awesome Lists containing this project

README

          

# xgboost-js

A pure JavaScript implementation of XGBoost for both Node.js and browser environments.

## Installation

```bash
npm install @fractal-solutions/xgboost-js
```

## Usage

```javascript
const { XGBoost } = require('@fractal-solutions/xgboost-js');

// Initialize the model
const model = new XGBoost({
learningRate: 0.3,
maxDepth: 4,
minChildWeight: 1,
numRounds: 100
});

// Train the model
model.fit(X_train, y_train);

// Make predictions
const predictions = model.predictBatch(X_test);
```

## Features

- Binary classification
- Model serialization
- Feature importance calculation
- Comprehensive test suite
- Pure JavaScript implementation
- Works in both Node.js and browser environments

## Documentation

For full documentation, see [docs/xgboost.md](docs/xgboost.md)

## License

MIT © Fractal Solutions