Commit Graph
26605 Commits
Author SHA1 Message Date
Rishabh Garg 4dc37b23a8 Changed labels to responses throughtout regression tree codebase 2021-07-12 10:32:56 +05:30
Rishabh Garg 26477c778c Removed unnecessary test 2021-07-12 10:32:56 +05:30
Rishabh Garg e0487772ea Added NumLeaves() to regression tree 2021-07-12 10:32:56 +05:30
Rishabh Garg b7d2f22bdc Add MultiSplit tests that ensure that tree learns obvious patterns in data 2021-07-12 10:32:56 +05:30
Rishabh Garg 8d8687a928 Add a mock categorical dataset and test DecisionTreeRegressor on it 2021-07-12 10:32:56 +05:30
Rishabh Garg 44c2b5ca51 Reverted AllCategoricalSplit to use arma::vec for splitInfo 2021-07-12 10:32:56 +05:30
Rishabh Garg 6e34bc02c6 Add random numeric splitting strategy for regression tree 2021-07-12 10:32:56 +05:30
Rishabh Garg fdb90df2d3 Reverted splitInfo to arma::vec for old decision tree for numric splits 2021-07-12 10:32:56 +05:30
Rishabh Garg 57572aa876 Removed numLeaves from BestBinaryNumericSplit 2021-07-12 10:32:56 +05:30
Rishabh Garg c174f68d0f Reorganised tests and added tests to ensure gain correctness by hand calculating the values 2021-07-12 10:32:56 +05:30
Rishabh Garg 271e90130e Weighted gain computation in BestBinaryNumericSplit 2021-07-12 10:32:56 +05:30
Rishabh Garg 8e4b891c82 fixed failing test after merging master 2021-07-12 10:32:56 +05:30
Rishabh Garg b5c72f1268 Fixed MSEgain computation 2021-07-12 10:28:54 +05:30
Rishabh Garg b6145adab8 Add hand created dataset for testing 2021-07-12 10:28:02 +05:30
Rishabh Garg aa78c26d7a Added more debugging code 2021-07-12 10:27:38 +05:30
Rishabh Garg 64fde38ae8 Testing for bug in implementation 2021-07-12 10:27:24 +05:30
Rishabh Garg c2c85f0250 Add initial tests 2021-07-12 10:26:18 +05:30
Rishabh Garg 4f000d5894 Add evaluation metrics for testing 2021-07-12 10:26:18 +05:30
Rishabh Garg a3fc8d9fe6 Add to CMakeLists.txt 2021-07-12 10:26:18 +05:30
Rishabh Garg be81c14a52 Implement DecisionTreeRegressor 2021-07-12 10:26:18 +05:30
Rishabh Garg dcc96c056c Amend RandomBinaryNumericSplit signature to support regression 2021-07-12 10:25:30 +05:30
Rishabh Garg 0449ace922 Update documentation 2021-07-12 10:16:31 +05:30
Rishabh Garg c6f0d29457 Make LabelsType a new template parameter and use subvec for labels 2021-07-12 10:16:31 +05:30
Rishabh Garg c4e3d1fcc3 Add TODO for optimization in BestBinaryNumericSplit for regression 2021-07-12 10:16:31 +05:30
Rishabh Garg ab4f3c3646 Added tests for best binary split 2021-07-12 10:13:35 +05:30
Rishabh Garg 536c5f9880 Implemented best binary split for regression 2021-07-12 10:13:35 +05:30
Rishabh Garg 7e336ad118 Fixed implementation bug in calculating sum of subvector 2021-07-12 10:12:57 +05:30
Rishabh Garg 722df85864 Refactored BestBinaryNumericSplit 2021-07-12 10:12:36 +05:30
Rishabh Garg 2d20ce9099 Added tests for AllCategoricalSplit for regression 2021-07-12 10:12:19 +05:30
Rishabh Garg f541fe20b0 Added ignored numClasses to MSE and MAD gains 2021-07-12 10:11:58 +05:30
Rishabh Garg ff85e20b96 Shifted regression tree tests to new file 2021-07-12 10:11:17 +05:30
Rishabh Garg 6e6d60f2b3 Improved documentation 2021-07-12 10:10:01 +05:30
Rishabh Garg e3579871af Refactored AllCategoricalSplit patch 2 2021-07-12 10:10:01 +05:30
Rishabh Garg 37559821bf Refactored AllCategoricalSplit 2021-07-12 10:10:01 +05:30
Rishabh Garg a2c9cd0c2d Added Train overloads 2021-07-12 10:10:01 +05:30
Rishabh Garg 0dd0715b5f Implemented ctors and dtor 2021-07-12 10:10:01 +05:30
Rishabh Garg 03a3f378d2 Added copy and move ctors and dtor 2021-07-12 10:10:01 +05:30
Rishabh Garg a030a556d5 Added constructors 2021-07-12 10:10:01 +05:30
Rishabh Garg 13311bb25c Added constructors 2021-07-12 10:10:01 +05:30
Rishabh Garg 2690c2c489 Removed documentation of ElemType 2021-07-12 10:08:37 +05:30
Rishabh Garg 415ac24c50 Removed print statement 2021-07-12 10:08:09 +05:30
Rishabh Garg b7d4c25c32 Implemented tests for MSE gain 2021-07-12 10:07:37 +05:30
Rishabh Garg be9ff3437e Implemented MSE gain 2021-07-12 10:06:25 +05:30
Rishabh Garg e47962f18c Refactored mad_gain to use utils.hpp 2021-07-12 10:06:00 +05:30
Rishabh Garg 24cd615fc6 Implemented SIMD sum in utils.hpp 2021-07-12 10:05:40 +05:30
Rishabh Garg d8634ca083 Fixed logic error in calculating weighted mean 2021-07-12 10:05:19 +05:30
Rishabh Garg 0c265ccffe Fixed implementation bug and normal distribution test 2021-07-12 10:05:01 +05:30
Rishabh Garg 8d3a7d7799 Added forgotten import and fixed test 2021-07-12 10:04:24 +05:30
Rishabh Garg 80a094e545 Added mad gain implementation and tests 2021-07-12 10:04:05 +05:30
Aakash kaushik 42071e92be Addition of ReLU6 (#3009)
* relu6 files added

* add into layer's cmake

* added tests

* applied suggestions

* comment about torch values in tests
2021-07-12 09:36:46 +05:30