Skip to content
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

Return fitted values during training phase #493

Closed
TiaHao opened this issue Apr 9, 2022 · 2 comments
Closed

Return fitted values during training phase #493

TiaHao opened this issue Apr 9, 2022 · 2 comments
Labels
feature request A tag for feature requests v2.0.0

Comments

@TiaHao
Copy link

TiaHao commented Apr 9, 2022

Is your feature request related to a problem? Please describe.

Hi,

This library is quite powerful and I appreciate it a lot. I am wondering is there any way to return back the fitted values for training phase, like the fittedvalues attribute of sarima model in statsmodels package? I would like to compute the R2, RMSE, etc for training phase as well.

Thank you!

Describe the solution you'd like

one added attribute for the train function

Describe alternatives you've considered

directly return back some evaluation metrics: R2, RMSE, MSE, MAPE etc. for training phase

Additional Context

No response

@TiaHao TiaHao added the feature request A tag for feature requests label Apr 9, 2022
@tgsmith61591
Copy link
Member

This can currently be accessed via the private arima_res_ attribute. But this would be a pretty easy attribute to expose at the top-level. We'll get this into the next release

import pmdarima as pm

y = pm.datasets.load_wineind()
mod = pm.ARIMA((1, 0, 2)).fit(y)
mod.arima_res_.fittedvalues

tgsmith61591 added a commit that referenced this issue Apr 16, 2022
tgsmith61591 added a commit that referenced this issue Apr 16, 2022
* Address #493

* Add a unit test

* Fix build requirements

* 📚 Add to doc. Again...
@tgsmith61591
Copy link
Member

Merged into master, and will be released in the next version

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature request A tag for feature requests v2.0.0
Projects
None yet
Development

No branches or pull requests

2 participants