Expand/change Ensemble to include AbsEndcap #55
Labels
good first issue
Good for newcomers
improvement
Something which would improve current status, but not add anything new
low priority
Not urgent and won't degrade with time
Current state
AbsEndcap
acts as a wrapper to apply fixed functions to the outputs of models that were trained on proxy objectives, e.g. to compute the invariant mass from a model that outputs the 3-momenta of two particles (see Multi_Target_Regression example).AbsEndcap
currently does this via a.forward
method and a.predict
method, so in theory, this should also work for Ensembles of models. Wrapping anEnsemble
, though, then removes many of the benefits of that theEnsemble
class offers.Proposal
The
Ensemble
class is extended to include the ability to add anAbsEndcap
, through which the outputs of all internal models will be passed.The text was updated successfully, but these errors were encountered: