Open kordc opened 1 year ago
We agreed on making just ensemble step
Deep learning ensembles' training is a huge topic. I found a nice package Ensembles-PyTorch. They provide a wide selection of already implemented ensemble types. I'll play around with this library using our ArtModule
Nice, if we won't need to write it by ourselves it would be great.
It supports only classification and regression. Even multi-label classification is not supported.
I think doing this step a task-agnostic is simply not possible. I see three options:
I think weighted mean might be good enough
user can specify weights
Same as @trebacz626 suggests. Any more sophisticated ensembles can be done by the user.
Actually, most tasks are classification or regression. If we can utilize this library quickly/easily lets use it, even though it doesn't support everything. If user wants something more sophisticated they must implement it by themselves.
I'd focus there on the 2 steps he tells us about:
So maybe instead of the squeeze, we should enable ensemble DL training?