JSMLT
The JavaScript Machine Learning Toolkit, or JSMLT, is an open source JavaScript library for education in machine learning. It implements several well-known supervised learning algorithms in an understandable, modular and well-commented way. Furthermore, visualization examples are provided which allow you to explore the way different machine learning algorithms work. Ultimately, JSMLT is intended to provide students with a better learning experience when studying machine learning algorithms.
If you want to explore a visualization of the machine learning algorithms in JSMLT, check out visualml.io. It provides an interactive environment for using JSMLT's algorithms.
Getting started
This short guide will help you get started with JSMLT.
Installation
We're assuming you've got Node.js and npm installed. If you haven't, you should: download and install it from nodejs.org.
To install JSMLT into your npm project via npm, run
npm install @jsmlt/jsmlt
A simple example
In this small example, we're going to train an SVM on a small example dataset. The code example below starts with loading JSMLT, creating some dummy training and test data, and running an SVM classifier on it. It's pretty simple!
If you want to run this example without having to set up anything by yourself, check out the JSMLT examples repository. It includes the example below, and requires no further setup: it's ready to run!
// Import JSMLT library
var jsmlt = require('@jsmlt/jsmlt');
// Training data
train_X = [[-1,-1], [-1,1], [1,1], [1,-1]];
train_y = [0, 0, 1, 1];
// Testing data
test_X = [[1,2], [1,-2], [-1,-2], [-1,2]];
// Create and train classifier
var clf = new jsmlt.Supervised.SVM.SVM({
kernel: new jsmlt.Kernel.Linear(),
});
clf.train(train_X, train_y);
// Make predictions on test data
console.log(clf.predict(test_X));
Running this simple example will output the classification result [1,1,0,0], meaning it classified the first two points as 0, and the second two points as 1.
API
The entire API documentation can be found here. You can also build the documentation locally by downloading and installing JSMLT and running
npm run-script build-documentation: the documentation will then be available in thedocsfolder.
Supervised learning algorithms (classifiers)
- Random Forest:
JSMLT.Supervised.Trees.RandomForest - Decision Tree:
JSMLT.Supervised.Trees.DecisionTree - Support Vector Machine (SVM):
JSMLT.Supervised.SVM.SVM - Perceptron:
JSMLT.Supervised.Linear.Perceptron - Neural Networks:
JSMLT.Supervised.NeuralNetwork.FullyConnected - k-nearest neighbors:
JSMLT.Supervised.Neighbors.KNN - Logistic Regression:
JSMLT.Supervised.Neighbors.LogisticRegression
Unsupervised learning algorithms (clustering)
- k-means:
JSMLT.Unsupervised.Neighbors.KMeans
Kernels
- Linear kernel:
JSMLT.Kernel.LinearKernel - Gaussian (RBF) kernel:
JSMLT.Kernel.GaussianKernel - Polynomial kernel:
JSMLT.Kernel.PolynomialKernel - Sigmoid kernel:
JSMLT.Kernel.SigmoidKernel
Preprocessing
- Encode string or other type of labels to integers:
JSMLT.Preprocessing.LabelEncoder
Model selection
- Data set splitting:
JSMLT.ModelSelection.trainTestSplit
Datasets
- Iris dataset loading:
JSMLT.Datasets.loadIris
Validation
- Accuracy metric for validation:
JSMLT.Validation.Metrics.accuracy - AUROC metric for validation:
JSMLT.Validation.Metrics.auroc
Classification boundaries
- Classification boundaries for trained classifier:
JSMLT.Classification.Boundaries
Development
JSMLT is maintained by Jesper van Engelen, and is in active development. It is currently not ready to be used in any production environments.
2.3k Jan 1, 2023
4.4k Jan 4, 2023
1k Dec 16, 2022
59 Nov 21, 2022
5.9k Jan 2, 2023
108 Feb 26, 2021
522 Jan 2, 2023
26 Oct 7, 2022
1.6k Dec 26, 2022
10.4k Dec 31, 2022
45 Mar 7, 2022
296 Nov 6, 2022
107 Oct 24, 2022
3 Jun 23, 2022
8k Dec 26, 2022
1.5k Dec 31, 2022
29.2k Dec 31, 2022
8.5k Dec 31, 2022
2.7k Dec 29, 2022