5 Javascript Libraries to Use for Machine Learning

5 Javascript Libraries to Use for Machine Learning

Danielle Maxwell | Monday, Mar 11, 2024 |  ML NLPNeural NetworksJavaScript

Over the years, several JavaScript libraries have been created for machine learning. Let’s sort through the ones that can help you get started quickly, even if you don’t have much experience with machine learning or data science.

Before getting to the list, I should note that it was difficult to narrow it down to 5. There were many libraries to choose from, but I wanted to focus on a diverse set of tools that I believe are also developer friendly.

TensorFlow.js

TensorFlow.js is quite possibly the most well known JavaScript machine learning library. First announced by Google in 2018, TensorFlow.js allows for the development of machine learning models in JavaScript via the browser or Node.js. Developers can work with existing models and also import their own.

If you’re in need of a little inspiration for working with TensorFlow.js, check out this playlist of what others have built with it.

Brain.js

Want to train your own model? Look no further than Brain.js. After installing the library, you can train a model in just a few lines of code. Once you have your dataset, follow one of their examples and get going in no time.

Brain.js allows for training data to be saved and serialized to JSON. This could be useful if you need to add data and retrain a model.

Danfo.js

This library was built to make data manipulation and processing easier in JavaScript. Danfo.js can be used in the browser, with client-side libraries such as React, and in Node.js environments.

Danfo.js was built on TensforFlow.js, which allows for dataframes to be converted into tensors and vice versa. Another benefit of using the library is how easy it is to remove or fill in missing data.

If you’re familiar with using Python’s pandas library, you may notice some similarities since Danfo.js was modeled after pandas.

ml.js

Need to work with classification and regression algorithms for your next machine learning project? ml.js has got you covered.

They’ve even compiled all of their machine learning and numerical analytics tools into the ml.js library which is currently recommended for use in the browser.

Should you need to work with one of their tools in Node.js, it is recommended to install dependencies individually. Check out all of their repositories here to get started.

winkJS

Though more of a collection of libraries, I could not leave out winkJS. This compilation of open source libraries makes it possible to easily work with Natural Language Processing (NLP), machine learning, and statistical analysis in JavaScript either in the browser or Node.js.

The most popular library, winkNLP

has a comprehensive natural language processing (NLP) pipeline covering tokenization, sentence boundary detection (sbd), negation handling, sentiment analysis, part-of-speech (pos) tagging, named entity recognition (ner), custom entities recognition (cer).

Another great thing about winkNLP is that it has full support for TypeScript.

Build Machine Learning Projects with JavaScript

I hope this list has inspired you to use one of or all of these tools while working on your next machine learning project. If there’s a JavaScript library that you think should be on this list, let me know!

Photo by Prateek Katyal on Unsplash

About This Post

Get to know 5 JavaScript libraries you can use for machine learning.

Written by:

Share this post:

Recent Rotations butterfly

View all

To LLM or Not to LLM (Part 2): Starting Simple

Sick of hearing about hyped up AI solutions that sound like hot air? 🧐 Let’s use boring old ML to detect hype in AI marketing text and see why starting with a simple ML approach is still your best bet 90% of the time.

Building an AI Text Detector - Lessons Learned

The LLMs boom has made differentiating text written by a person vs. generated by AI a highly desired technology. In this post, I’ll attempt to build an AI text detector from scratch!

May 15, 2024

To LLM or Not to LLM: Tips for Responsible Innovation

We’re seeing a proliferation of Large Language Models (LLMs) as companies seek to replicate OpenAI’s success. In this post, two AI engineers respond to LLM FAQs and offer tips for responsible innovation.

Enter Your Email To Subscribe