Fixed issues 9, 15, 17, 28, 32

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 #34

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

aiday-mar commented 2 years ago

I have added a scripts folder as suggested by Prof. Patiny which contains code I wrote in order to test and fix the issues. This folder can be deleted as it is there for testing purposes.