Skip to content

Commit 001fc12

Browse files
authored
feat: add ml-cart to the package (#93)
1 parent c6dd927 commit 001fc12

File tree

4 files changed

+21
-0
lines changed

4 files changed

+21
-0
lines changed

README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ It will be available as the global `ML` variable. The package is in UMD format a
3131
* Partial least squares (PLS): [`ML.PLS`](https://github.com/mljs/pls)
3232
* Cross-validation: [`ML.CrossValidation`](https://github.com/mljs/cross-validation)
3333
* Confusion matrix: [`ML.ConfusionMatrix`](https://github.com/mljs/confusion-matrix)
34+
* Decision tree classifier: [`ML.DecisionTreeClassifier`](https://github.com/mljs/decision-tree-cart)
3435

3536
### Artificial neural networks (ANN)
3637

@@ -46,6 +47,7 @@ It will be available as the global `ML` variable. The package is in UMD format a
4647
* Exponential regression: [`ML.ExponentialRegression`](https://github.com/mljs/regression-exponential)
4748
* Theil-Sen regression: [`ML.TheilSenRegression`](https://github.com/mljs/regression-theil-sen)
4849
* Robust polynomial regression: [`ML.RobustPolynomialRegression`](https://github.com/mljs/regression-robust-polynomial)
50+
* Decision tree regression: [`ML.DecisionTreeRegression`](https://github.com/mljs/decision-tree-cart)
4951

5052
### Optimization
5153

package-lock.json

Lines changed: 14 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

package.json

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@
3535
"binary-search": "^1.3.2",
3636
"ml-array-utils": "^0.4.0",
3737
"ml-bit-array": "^1.0.0",
38+
"ml-cart": "^1.0.8",
3839
"ml-confusion-matrix": "^0.3.0",
3940
"ml-cross-validation": "^1.1.0",
4041
"ml-distance": "^2.1.1",

src/index.js

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,7 @@
11
'use strict';
22

3+
const mlCart = require('ml-cart');
4+
35
const ML = exports;
46

57
// Try to keep this list in the same structure as the README.
@@ -16,6 +18,7 @@ ML.KNN = require('ml-knn');
1618
ML.PLS = require('ml-pls');
1719
ML.CrossValidation = require('ml-cross-validation');
1820
ML.ConfusionMatrix = require('ml-confusion-matrix');
21+
ML.DecisionTreeClassifier = mlCart.DecisionTreeClassifier;
1922

2023
// Artificial neural networks
2124
ML.FNN = require('ml-fnn');
@@ -30,6 +33,7 @@ ML.PowerRegression = Regression.PowerRegression;
3033
ML.ExponentialRegression = Regression.ExponentialRegression;
3134
ML.TheilSenRegression = Regression.TheilSenRegression;
3235
ML.RobustPolynomialRegression = Regression.RobustPolynomialRegression;
36+
ML.DesisionTreeRegression = mlCart.DesisionTreeRegression;
3337

3438
// Optimization
3539
ML.levenbergMarquardt = require('ml-levenberg-marquardt');

0 commit comments

Comments
 (0)