search

NVIDIA-Merlin / models

Merlin Models is a collection of deep learning recommender system model reference implementations
https://nvidia-merlin.github.io/models/main/index.html
Apache License 2.0
262 stars 50 forks source link

[Task] Generalize TopK to all model types doing multiclass classification in Merlin Models #1050

Open EvenOldridge opened 1 year ago

EvenOldridge commented 1 year ago

add a topK operation on top of softmax layer.