DFKDE refactor to accept any layer name like other models

[ashishbdatta]

Description

Slight refactor of the DFKDE model to accept from the config file the name of the layer needed. Currently this is hard coded to avgpool but to be consistent with other models will need to make this configurable.

Changes

• Bug fix (non-breaking change which fixes an issue)
• Refactor (non-breaking change which refactors the code base)
• New feature (non-breaking change which adds functionality)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)
• This change requires a documentation update

Checklist

• My code follows the pre-commit style and check guidelines of this project.
• I have performed a self-review of my code
• I have commented my code, particularly in hard-to-understand areas
• I have made corresponding changes to the documentation
• My changes generate no new warnings
• I have added tests that prove my fix is effective or that my feature works
• New and existing tests pass locally with my changes

[samet-akcay]

Thanks, I only have a single comment.