search

mljs / random-forest

Random forest for classification and regression.
https://mljs.github.io/random-forest/
MIT License
61 stars 21 forks source link

Fixed issues 9, 15, 17, 28, 32 #35

Closed aiday-mar closed 2 years ago

aiday-mar commented 2 years ago

The variable this.gainThreshold in the TreeNode implementation was not initialized properly and this was leading to an incorrect creation of regression tree. This will be adressed in a separate pull request for the ml-cart module.

Added the possiblity to change the number of samples one wants to use in the training phase

Added a function featureImportance in order to compute the feature importance as was suggested by @xdadda