Skip to content

Add RegressionOutput#1115

Merged
marcromeyn merged 7 commits intoNVIDIA-Merlin:mainfrom
edknv:torch/regression_output
May 29, 2023
Merged

Add RegressionOutput#1115
marcromeyn merged 7 commits intoNVIDIA-Merlin:mainfrom
edknv:torch/regression_output

Conversation

@edknv
Copy link
Contributor

@edknv edknv commented May 26, 2023

This PR introduces the RegressionOutput class, a prediction block for regression tasks, for the pytorch backend.

Depends on #1099.

@marcromeyn marcromeyn added area/pytorch enhancement New feature or request labels May 26, 2023
@marcromeyn marcromeyn added this to the Merlin 23.06 milestone May 26, 2023
@github-actions
Copy link

Documentation preview

https://nvidia-merlin.github.io/models/review/pr-1115

@edknv edknv marked this pull request as ready for review May 26, 2023 16:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area/pytorch enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants