From 76723993fdea8401955d16deb65abd90d2df0ab6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Montan=CC=83ana?= <rmontanana@gmail.com>
Date: Thu, 25 Jun 2020 13:07:50 +0200
Subject: [PATCH] Solve Warning class label not found when bagging

---
 stree/Strees.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/stree/Strees.py b/stree/Strees.py
index dc95247..7c32d32 100644
--- a/stree/Strees.py
+++ b/stree/Strees.py
@@ -440,7 +440,12 @@ class Stree(BaseEstimator, ClassifierMixin):
 
         check_classification_targets(y)
         X, y = check_X_y(X, y)
-        sample_weight = _check_sample_weight(sample_weight, X)
+        sample_weight = _check_sample_weight(
+            sample_weight, X, dtype=np.float64
+        )
+        # solve WARNING: class label 0 specified in weight is not found
+        # in bagging
+        sample_weight += 1e-5
         check_classification_targets(y)
         # Initialize computed parameters
         self.splitter_ = Splitter(