Commit Graph

4 Commits

Author SHA1 Message Date
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