Add the probabilities aggregation method to compute prediction with ensembles #16

rmontanana · 2024-02-25T09:54:24Z

rmontanana commented

2024-02-25 09:54:24 +00:00

As only the voting method was implemented, this approach computes the classifiers prediction using a weighted average of the probabilities computed by each model.
Added the predict_proba methods to BaseClassifier - Classifier and Ensemble classes.
Add a hyperparameter to decide the type of computation for ensembles voting - probability aggregation

As only the voting method was implemented, this approach computes the classifiers prediction using a weighted average of the probabilities computed by each model. Added the predict_proba methods to BaseClassifier - Classifier and Ensemble classes. Add a hyperparameter to decide the type of computation for ensembles voting - probability aggregation

rmontanana added 7 commits 2024-02-25 09:54:25 +00:00

Add predict_voting and predict_prob to ensemble e1c4221c11

Implement classifier.predict_proba & test 443e5cc882

Begin testing ensemble predict_proba 3116eaa763

Implement the proba branch and begin with the voting one 52abd2d670

Complete predict & predict_proba with voting & probabilities 8477698d8d

Complete predict & predict_proba in ensemble 02e456befb