Skip to content
Snippets Groups Projects
Unverified Commit 863032ee authored by André Pedersen's avatar André Pedersen Committed by GitHub
Browse files

feat: Support for setting aggregation for (Hybrid)RouteLayer (#202)


* Enable to set top_k for RouteLayer

* Add top_k unit test for RouteLayer

* Added support for setting different aggregation method for HybridRL

* Add unit test to verify that aggregation works

* Add support for setting aggregation in RouteLayer

* Fix bug where agg method not used in RL

* Add tests for agg

* Linted code

* Updated codecov file

* Change aggregation method names to lower case

---------

Co-authored-by: default avatarJames Briggs <35938317+jamescalam@users.noreply.github.com>
parent 1394adf5
No related branches found
No related tags found
No related merge requests found
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment