Commit Graph

18 Commits

Author SHA1 Message Date
502ee72799 #2 Add predict and score support
Add a test in features notebook
Show max_features in main.py
2020-06-14 14:00:21 +02:00
f1ee4de37b #2 - Add gini and entropy measures
rename get_dataset to load_dataset
add features and impurity to  __str__ of node
2020-06-14 03:08:55 +02:00
f360a2640c #6 - Add multiclass support
Removed (by now) predict_proba. Created a notebook in jupyter
Added split_criteria parameter with min_distance and max_samples values
Refactor _distances
Refactor _split_criteria
Refactor _reorder_results
2020-06-11 13:10:52 +02:00
b9f14aec05 #4 Add code coverage & codacy badge
Add code coverage configuration in codecov
Add some tests
2020-06-06 03:04:18 +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
6e35628c85 Grapher working 2020-05-20 14:26:55 +02:00
6ebd0f9be3 integrate iterator in Stree 2020-05-19 18:19:23 +02:00
68512b3d75 test results & fix single predict_proba 2020-05-18 17:08:05 +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
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
e3ae3a3a6c Add C param in constructor and creditcard dataset 2020-05-14 10:48:39 +02:00
8f71eeb316 Implement predict and score methods & tests 2020-05-13 12:42:09 +02:00
c4de782a3f compute predictor and store model in node 2020-05-13 00:12:05 +02:00
371257c121 Implement split data with or without using predictions & some tests 2020-05-12 17:40:06 +02:00
b9ea24696f Working tree with samples and first test 2020-05-10 14:25:58 +02:00