Doctorado-ML/stree

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

Go to file

Ricardo Montañana 5e5fea9c6a

Document & lint code

2020-05-30 23:10:10 +02:00

Document & lint code

2020-05-30 23:10:10 +02:00

Document & lint code

2020-05-30 23:10:10 +02:00

.gitignore

compute predictor and store model in node

2020-05-13 00:12:05 +02:00

.travis.yml

Make project python package friendly

2020-05-23 23:40:33 +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

Make project python package friendly

2020-05-23 23:40:33 +02:00

README.md

Adapt some notebooks

2020-05-30 11:09:59 +02:00

requirements.txt

update requirements to alllow maptlot widget

2020-05-23 00:05:58 +02:00

setup.py

Document & lint code

2020-05-30 23:10:10 +02:00

README.md

Stree

Oblique Tree classifier based on SVM nodes. The nodes are built and splitted with sklearn LinearSVC 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

Slow launch but better integration

Test notebook

Fast launch but have to run first commented out cell for setup

Test notebook
Another Test notebook
Test Graphics notebook

Command line

python main.py

Tests

python -m unittest -v stree.tests