Merge branch 'master' of github.com:Doctorado-ML/odte

This commit is contained in:
Ricardo Montañana Gómez 2025-05-20 11:42:39 +02:00
commit 0f4ea8542e
Signed by: rmontanana
GPG Key ID: 46064262FD9A7ADE
2 changed files with 4 additions and 4 deletions

View File

@ -1,3 +1,5 @@
# Odte
![CI](https://github.com/Doctorado-ML/Odte/workflows/CI/badge.svg)
[![CodeQL](https://github.com/Doctorado-ML/Odte/actions/workflows/codeql-analysis.yml/badge.svg)](https://github.com/Doctorado-ML/Odte/actions/workflows/codeql-analysis.yml)
[![codecov](https://codecov.io/gh/Doctorado-ML/odte/branch/master/graph/badge.svg)](https://codecov.io/gh/Doctorado-ML/odte)
@ -6,6 +8,4 @@
![https://img.shields.io/badge/python-3.11%2B-blue](https://img.shields.io/badge/python-3.11%2B-brightgreen)
[![DOI](https://zenodo.org/badge/271595804.svg)](https://zenodo.org/badge/latestdoi/271595804)
# Odte
Oblique Decision Tree Ensemble classifier based on [STree](https://github.com/doctorado-ml/stree) nodes.

View File

@ -272,6 +272,6 @@ class Odte(BaseEnsemble, ClassifierMixin):
check_is_fitted(self, "estimators_")
return self.depth_
def nodes_leaves(self) -> Tuple[float, float]:
def nodes_leaves(self) -> Tuple[int, int]:
check_is_fitted(self, "estimators_")
return self.nodes_, self.leaves_
return (self.get_nodes(), self.get_leaves())