-
Notifications
You must be signed in to change notification settings - Fork 48
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
@@ -119,6 +119,35 @@ def predict( | |||
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
@@ -186,6 +186,11 @@ def arima_evaluate(self, show_all_candidate_models: bool = False): | |||
|
|||
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> 🦕