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 ...

Get Learning Jupyter now with the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.