scikit-learn-contrib / DESlib

A Python library for dynamic classifier and ensemble selection
BSD 3-Clause "New" or "Revised" License
477 stars 106 forks source link

Hard soft voting #248

Closed Menelau closed 3 years ago

Menelau commented 3 years ago

Adding Hard and Soft voting options for dynamic ensemble selection (DES) models (Fixes #133). This PR also fixes the inconsistency in the implementations of predict and predict_proba functions which were using hard and soft voting respectively and could lead to different results.

New implementation allows users to select the voting type by setting the voting argument.

Soft voting requires that each base model can estimate probabilities (implements the predict_proba function). Each base model should be well-calibrated for better performance.