Commit Graph

11 Commits

Author SHA1 Message Date
ed98054f0d First approach
Added max_depth, tol, weighted samples
2020-05-29 12:46:10 +02:00
77f10281c1 Make project python package friendly
- Add setup.py
- Move classes to module files
- Move tests folder inside module folder
2020-05-23 23:40:33 +02:00
e51690ed95 Implement grapher and notebook to test it 2020-05-22 19:42:13 +02:00
a4595f5815 Update notebooks and readme with cosmetic changes 2020-05-20 18:11:57 +02:00
9f30627e47 Fix main and small issues in notebook test 2020-05-17 18:06:03 +02:00
d997bc3460 Add model comparison to notebook test 2020-05-17 16:17:57 +02:00
ad51d0b36b Update readme, add travis config 2020-05-17 00:13:29 +02:00
6a8ef288ce add _linear_function to Stree
First try to implement predict_proba with sigmoid based on Platt
2020-05-16 20:34:03 +02:00
80b5cf8e72 Implement predict & predict_proba optimization
reduces time in two orders of magnitude in creditcard dataset
2020-05-15 23:35:33 +02:00
e56b955b92 Implement predict_proba with test.
Fix tree overload with dataset in nodes only needed in tests
2020-05-14 18:42:17 +02:00
Ricardo Montañana Gómez
2ca25fd656 Initial commit 2020-05-09 23:01:30 +02:00