Skip to content

How to get un-normalized regression predictions #99

Answered by manujosephv
Aceticia asked this question in Q&A
Discussion options

You must be logged in to vote

PyTorch Tabular does not use any target transformation by default. There is a way to use a target_transformer in the fit method, but if you do that, they are reverse transformed before reporting the predictions.

What might be happening in your case can be that the model might be struggling to capture the scale of the data. There are two things you can try:

  1. Use target_range parameter in ModelConfig and specify the range you expect your output to be in. This uses a sigmoid layer to force the output to be in between the given range. But this is only a good way if you expect something similar to a normal distribution for your target.
  2. Scale the target. Even something as simple as dividing by …

Replies: 1 comment 3 replies

Comment options

You must be logged in to vote
3 replies
@miaozi-hao
Comment options

@manujosephv
Comment options

@miaozi-hao
Comment options

Answer selected by Aceticia
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
3 participants