mirror of
https://github.com/Doctorado-ML/STree.git
synced 2025-08-15 15:36:00 +00:00
d7c0bc3bc5b871e560209a0e484a754815035424
Add multiclass dimensions management in distances method Add gamma hyperparameter for non linear kernels
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%