-
Notifications
You must be signed in to change notification settings - Fork 63
feat: add ARIMAPlus.coef_ property exposing ML.ARIMA_COEFFICIENTS functionality
#585
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
Conversation
…uery-dataframes into salem_timeseriessample
…es into salem_timeseriessample
bigframes/ml/forecasting.py
Outdated
| options={"horizon": horizon, "confidence_level": confidence_level} | ||
| ) | ||
|
|
||
| def arima_coefficients( |
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 discussed exposing this as an @property called coef_ to match sklearn.
bigframes/ml/core.py
Outdated
|
|
||
| return self._session.read_gbq(sql) | ||
|
|
||
| def coef_(self) -> bpd.DataFrame: |
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.
Let's use the SQL name here too.
ARIMAPlus.coef_ property exposing ML.ARIMA_COEFFICIENTS functionality
|
FYI: I updated the PR title to "feat: add Looks like the relevant tests are passing. :-) http://fusion2/ci/kokoro/prod%3Abigframes%2Fpresubmit%2Fe2e/activity/4cfd1e01-59bb-4955-8780-40c7fc914720 @GarrettWu @ashleyxuu The reason |
Thank you for opening a Pull Request! Before submitting your PR, there are a few things you can do to make sure it goes smoothly:
Fixes #<issue_number_goes_here> 🦕