From 582ff44cb131de52da48a830a4a9058ab0767031 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?= Date: Wed, 9 Apr 2025 11:19:37 +0200 Subject: [PATCH] Update return types in nodes_leaves --- README.md | 4 ++-- odte/Odte.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 5011a65..bb56d53 100644 --- a/README.md +++ b/README.md @@ -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. diff --git a/odte/Odte.py b/odte/Odte.py index a48bc26..5d0b7af 100644 --- a/odte/Odte.py +++ b/odte/Odte.py @@ -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())