Home

Awesome

machinelearn.js

machinelearn.js is a Machine Learning library written in Typescript. It solves Machine Learning problems and teaches users how Machine Learning algorithms work.

Build Status Build status FOSSA Status Slack ZenHub

<img src="https://i.imgur.com/I5VbqB1.jpg">

User Installation

Using yarn

$ yarn add machinelearn

Using NPM

$ npm install --save machinelearn

On the browsers

We use jsdeliver to distribute browser version of machinelearn.js

<script src="https://cdn.jsdelivr.net/npm/machinelearn/machinelearn.min.js"></script>
<script>
    const { RandomForestClassifier } = ml.ensemble;
    const cls = new RandomForestClassifier();
</script>

Please see https://www.jsdelivr.com/package/npm/machinelearn for more details.

Accelerations

By default, machinelearning.js will use pure Javascript version of tfjs. To enable acceleration through C++ binding or GPU, you must import machinelearn-node for C++ or machinelearn-gpu for GPU.

  1. C++
yarn add machinelearn-node
import 'machinelearn-node';
  1. GPU
yarn add machinelearn-gpu
import 'machinelearn-gpu';

Highlights

Development

We welcome new contributors of all level of experience. The development guide will be added to assist new contributors to easily join the project.

Simplicity

machinelearn.js provides a simple and consistent set of APIs to interact with the models and algorithms. For example, all models have follow APIs:

Testing

Testing ensures you that you are currently using the most stable version of machinelearn.js

$ npm run test

Supporting

Simply give us a :star2: by clicking on <img width="45" src="https://i.imgur.com/JEOaKBk.png">

Contributing

We simply follow "fork-and-pull" workflow of Github. Please read CONTRIBUTING.md for more detail.

Further notice

Great references that helped building this project!

Contributors

Thanks goes to these wonderful people (emoji key):

<!-- ALL-CONTRIBUTORS-LIST:START - Do not remove or modify this section --> <!-- prettier-ignore -->
<img src="https://avatars0.githubusercontent.com/u/2525002?v=4" width="100px;"/><br /><sub><b>Jason Shin</b></sub><br />📝 🐛 💻 📖 ⚠️<img src="https://avatars0.githubusercontent.com/u/21098186?v=4" width="100px;"/><br /><sub><b>Jaivarsan</b></sub><br />💬 🤔 📢<img src="https://avatars2.githubusercontent.com/u/9072266?v=4" width="100px;"/><br /><sub><b>Oleg Stotsky</b></sub><br />🐛 💻 📖 ⚠️<img src="https://avatars3.githubusercontent.com/u/687794?v=4" width="100px;"/><br /><sub><b>Ben</b></sub><br />💬 🎨 📢 🐛 💻<img src="https://avatars1.githubusercontent.com/u/7292257?v=4" width="100px;"/><br /><sub><b>Christoph Reinbothe</b></sub><br />💻 🤔 🚇 👀<img src="https://avatars1.githubusercontent.com/u/14098106?v=4" width="100px;"/><br /><sub><b>Adam King</b></sub><br />💻 ⚠️ 📖
<!-- ALL-CONTRIBUTORS-LIST:END -->