Home > Blockchain >  Is it possible to average the output of multiple classification models using pipeline in sklearn?
Is it possible to average the output of multiple classification models using pipeline in sklearn?

Time:05-15

As an example, suppose there is a random forest and a logistic regression model that accept the same input data, and I want the inference result to be the average of the probabilities of these two models.

In this case, is it possible to create a pipeline that outputs the average probabilities of the two models as shown above?

CodePudding user response:

A VotingClassifier with voting="soft" will work for this purpose.

  • Related