Node.js decision-tree package
The decision-tree
package is an example of a machine learning package. It is available at
https://www.npmjs.com/package/decision-tree
. The package is installed using the following command:
npm install decision-tree
We need a dataset to use for training/developing our decision tree. I am using the car MPG dataset on this page: https://alliance.seas.upenn.edu/~cis520/wiki/index.php?n=Lectures.DecisionTrees . It did not seem to be available directly, so I copied it into Excel and saved it as a local CSV.
The logic for machine learning is very similar:
Load our dataset
Split into a training set and a testing set
Use the training set to develop our model
Test the mode on the test set.
Tip
Typically, you might use two-thirds of your data for training and one-third for testing.
Using the decision-tree
package and the car MPG dataset we would have a script similar to the following:
//Import the modules var DecisionTree = require('decision-tree'); var fs = require("fs");...