From b044a057df3bbc60476236c143e2e1824766db52 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= Date: Tue, 2 Nov 2021 14:04:10 +0100 Subject: [PATCH] Update comments and README.md --- README.md | 4 ++++ stree/Splitter.py | 3 ++- stree/Strees.py | 5 +++-- 3 files changed, 9 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 665c77d..bee1822 100644 --- a/README.md +++ b/README.md @@ -73,3 +73,7 @@ python -m unittest -v stree.tests ## License STree is [MIT](https://github.com/doctorado-ml/stree/blob/master/LICENSE) licensed + +## Reference + +R. Montañana, J. A. Gámez, J. M. Puerta, "STree: a single multi-class oblique decision tree based on support vector machines.", 2021 LNAI 12882, pg. 54-64 diff --git a/stree/Splitter.py b/stree/Splitter.py index acf737e..5c78775 100644 --- a/stree/Splitter.py +++ b/stree/Splitter.py @@ -202,7 +202,8 @@ class Splitter: max_features < num_features). Supported strategies are: “best”: sklearn SelectKBest algorithm is used in every node to choose the max_features best features. “random”: The algorithm generates 5 candidates and - choose the best (max. info. gain) of them. "mutual": Chooses the best + choose the best (max. info. gain) of them. “trandom”: The algorithm + generates only one random combination. "mutual": Chooses the best features w.r.t. their mutual info with the label. "cfs": Apply Correlation-based Feature Selection. "fcbf": Apply Fast Correlation- Based, by default None diff --git a/stree/Strees.py b/stree/Strees.py index a17bbc6..1857bca 100644 --- a/stree/Strees.py +++ b/stree/Strees.py @@ -82,7 +82,8 @@ class Stree(BaseEstimator, ClassifierMixin): max_features < num_features). Supported strategies are: “best”: sklearn SelectKBest algorithm is used in every node to choose the max_features best features. “random”: The algorithm generates 5 candidates and - choose the best (max. info. gain) of them. "mutual": Chooses the best + choose the best (max. info. gain) of them. “trandom”: The algorithm + generates only one random combination. "mutual": Chooses the best features w.r.t. their mutual info with the label. "cfs": Apply Correlation-based Feature Selection. "fcbf": Apply Fast Correlation- Based , by default "random" @@ -128,7 +129,7 @@ class Stree(BaseEstimator, ClassifierMixin): References ---------- R. Montañana, J. A. Gámez, J. M. Puerta, "STree: a single multi-class - oblique decision tree based on support vector machines.", 2021 LNAI... + oblique decision tree based on support vector machines.", 2021 LNAI 12882 """