mirror of https://github.com/Doctorado-ML/STree.git synced 2025-08-15 23:46:02 +00:00

Go to file

rename get_dataset to load_dataset
add features and impurity to  __str__ of node

2020-06-14 03:08:55 +02:00

notebooks

Fix problem in _min_distance

2020-06-12 00:50:25 +02:00

stree

#2 - Add gini and entropy measures

2020-06-14 03:08:55 +02:00

.coveragerc

#4 Add code coverage & codacy badge

2020-06-06 03:04:18 +02:00

.gitignore

#4 Add code coverage & codacy badge

2020-06-06 03:04:18 +02:00

codecov.yml

#6 - Update tests and codecov conf

2020-06-11 13:45:24 +02:00

example.png

Implement grapher and notebook to test it

2020-05-22 19:42:13 +02:00

LICENSE

Initial commit

2020-05-09 23:01:30 +02:00

main.py

#2 - Add gini and entropy measures

2020-06-14 03:08:55 +02:00

pyproject.toml

#4 Add code coverage & codacy badge

2020-06-06 03:04:18 +02:00

README.md

#3 Add sample_weights to score, update notebooks

2020-06-09 01:46:38 +02:00

requirements.txt

Fix problem in _min_distance

2020-06-12 00:50:25 +02:00

setup.py

Fix problem in _min_distance

2020-06-12 00:50:25 +02:00

README.md

Stree

Oblique Tree classifier based on SVM nodes. The nodes are built and splitted with sklearn SVC models. Stree is a sklearn estimator and can be integrated in pipelines, grid searches, etc.

Installation

pip install git+https://github.com/doctorado-ml/stree

Examples

Jupyter notebooks

Benchmark
Benchmark
Test features
Adaboost
Gridsearch
Test Graphics

Command line

python main.py

Tests

python -m unittest -v stree.tests