Update return types in nodes_leaves

This commit is contained in:
Ricardo Montañana Gómez 2025-04-09 11:19:37 +02:00
parent 66fe0bbe48
commit 582ff44cb1
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) ![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) [![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) [![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) ![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) [![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. 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_") check_is_fitted(self, "estimators_")
return self.depth_ return self.depth_
def nodes_leaves(self) -> Tuple[float, float]: def nodes_leaves(self) -> Tuple[int, int]:
check_is_fitted(self, "estimators_") check_is_fitted(self, "estimators_")
return self.nodes_, self.leaves_ return (self.get_nodes(), self.get_leaves())