-
Notifications
You must be signed in to change notification settings - Fork 136
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
ktr lite house keeping #440
Conversation
- revert level knots plot - refine tutorial - refine some plotting code
- [x] enhance unit test - [x] relable pool - [x] initializer of ktrlite
x for x in self._level_knot_dates if | ||
(x <= df[self.date_col].max()) and (x >= df[self.date_col].min()) | ||
x for x in self._level_knot_dates if | ||
(x <= df[self.date_col].values[-1]) and (x >= df[self.date_col].values[0]) |
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.
this is assuming that df
is sorted by date_col
, though this should be a requirement for our model.
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.
just curious if there is any special consideration in this change?
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 have some impression that is what we check by default in the see _validate_training_df
under base template.
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.
LGTM!
* ktr lite house keeping - revert level knots plot - refine tutorial - refine some plotting code * more house keeping - [x] enhance unit test - [x] relable pool - [x] initializer of ktrlite * notebook update * Update test_ktrlite.py * Update test_ktrlite.py * add comment for stan_extract cleaning Co-authored-by: Zhishi Wang <zhishiw@uber.com>
* ktr lite house keeping - revert level knots plot - refine tutorial - refine some plotting code * more house keeping - [x] enhance unit test - [x] relable pool - [x] initializer of ktrlite * notebook update * Update test_ktrlite.py * Update test_ktrlite.py * add comment for stan_extract cleaning Co-authored-by: Zhishi Wang <zhishiw@uber.com>
* Prediction Plot Bug Fix and Enhance (#428) * Update plot.py * fix bug and refine prediction plot * Update quick_start.ipynb * Update plot.py * Predicion refactor (#430) * resolve conflicts * resolve conflicts * rebase commit * Prediction Plot Bug Fix and Enhance (#428) * Update plot.py * fix bug and refine prediction plot * Update quick_start.ipynb * Update plot.py * rebase commit * Refine KTRLite plotting and tutorial (#431) * rebase from dev * temp * Update build_your_own_model.ipynb * refine tutorials and ktrlite knots plotting * Update ktrlite.ipynb * palette and docstring * more verbiage Co-authored-by: Zhishi Wang <zhishiw@uber.com> * Update build_your_own_model.ipynb * Update build_your_own_model.ipynb * Feat arviz (#433) * arviz * get_posterior_samples * arviz plotting * arviz demo * rename * fix kwargs * refine plotting * more robust way to extract signatures of orbit models (#438) * more robust way to extract signatures of orbit models * unit test for grid search * replace regressor beta prior tuning with other params * ktr lite house keeping (#440) * ktr lite house keeping - revert level knots plot - refine tutorial - refine some plotting code * more house keeping - [x] enhance unit test - [x] relable pool - [x] initializer of ktrlite * notebook update * Update test_ktrlite.py * Update test_ktrlite.py * add comment for stan_extract cleaning Co-authored-by: Zhishi Wang <zhishiw@uber.com> * ELBO Loss Extraction (#443) * refine estimator to include training metrics for NUTS and SVI * Update pyro_basic.ipynb * add comment for stan_extract cleaning Co-authored-by: Zhishi Wang <zhishiw@uber.com> Co-authored-by: Zhishi Wang <wangzhishi@users.noreply.github.com> Co-authored-by: Zhishi Wang <zhishiw@uber.com> * use newly written arviz wrappers for diagnostic tutorial (#446) * resolve conflicts * rebase * rebase with dev * Update template.py Co-authored-by: Zhishi Wang <zhishiw@uber.com> Co-authored-by: Zhishi Wang <wangzhishi@users.noreply.github.com>
Description
It seems the older version of plot is more informative. I'm proposing to revert it with some better implementation.
Fixes #441