-
Notifications
You must be signed in to change notification settings - Fork 12
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Added normalization for predictions. #91
base: main
Are you sure you want to change the base?
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The aleatoric uncertainties would also need to be denormed
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think line 108 handles that case as well, as in case the model is robust, preds
will contain both (line 113) -
preds, aleat_log_std = preds.T
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Hi, I understand the problem you are pointing out, have added a fix.
aviary/predict.py
Outdated
# denorm the mean and aleatoroc uncertainties separately | ||
mean, log_std = np.split(preds, 2, axis=1) | ||
preds = normalizer.denorm(mean) | ||
ale_std = np.exp(log_std) * normalizer.std |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
we need to put this back to the log space here based on the logic below.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I would think it would be less code to just add the normalizer into the logic below rather than having to make a new logic block.
Hi, I agree with both of your suggestions. I made the changes. |
Fix for issue #90. I have only added a few lines of code in
predict.py
so that Roost models can be used for prediction later.