Skip to content

Commit

Permalink
feat: Support for setting aggregation for (Hybrid)RouteLayer (#202)
Browse files Browse the repository at this point in the history
* 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: James Briggs <35938317+jamescalam@users.noreply.github.com>
  • Loading branch information
andreped and jamescalam committed Mar 15, 2024
1 parent 1394adf commit 863032e
Show file tree
Hide file tree
Showing 5 changed files with 344 additions and 183 deletions.
Loading

0 comments on commit 863032e

Please sign in to comment.