mirror of
https://github.com/Doctorado-ML/STree.git
synced 2025-08-15 15:36:00 +00:00
3a48d8b40512d3259faf7238359fa29669db60dd
Remove use_predictions parameter as of now, the model always use it
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
Slow launch but better integration
Fast launch but have to run first commented out cell for setup
Command line
python main.py
Tests
python -m unittest -v stree.tests
Languages
Python
72.2%
Jupyter Notebook
26.9%
Makefile
0.9%