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

Go to file

Ricardo Montañana Gómez e4ac5075e5 Add main workflow action (#20 )

* Add main workflow action

* lock scikit-learn version to 0.23.2

* exchange codeship badge with githubs

2021-01-11 13:46:30 +01:00

.github/workflows

Add main workflow action (#20 )

2021-01-11 13:46:30 +01:00

notebooks

Enhance partition (#16 )

2020-11-03 11:36:05 +01:00

stree

Combinatorial explosion (#19 )

2021-01-10 13:32:22 +01:00

.coveragerc

better solution to the sklearn bagging problem

2020-06-26 11:22:45 +02:00

.gitignore

Get only 3 sets for best split

2020-06-27 18:29:40 +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 predict and score support

2020-06-14 14:00:21 +02:00

pyproject.toml

#4 Add code coverage & codacy badge

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

README.md

Add main workflow action (#20 )

2021-01-11 13:46:30 +01:00

requirements.txt

Add main workflow action (#20 )

2021-01-11 13:46:30 +01:00

setup.py

Add main workflow action (#20 )

2021-01-11 13:46:30 +01: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