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 by 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 from the following web 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
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...