Doctorado-ML/stree
1
1
Fork 0
mirror of https://github.com/Doctorado-ML/STree.git synced 2025-08-16 07:56:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Solve Warning class label not found when bagging

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2020-06-25 13:07:50 +02:00
parent ecd0b86f4d
commit 76723993fd
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
stree/Strees.py
View File

@@ -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(