-
Notifications
You must be signed in to change notification settings - Fork 917
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
Feat/encoders extension #1093
Feat/encoders extension #1093
Conversation
Codecov Report
@@ Coverage Diff @@
## master #1093 +/- ##
==========================================
+ Coverage 93.71% 93.72% +0.01%
==========================================
Files 80 80
Lines 8236 8250 +14
==========================================
+ Hits 7718 7732 +14
Misses 518 518
Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here. |
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 a great improvement 🚀 😍
Thanks!
@@ -59,14 +51,10 @@ | |||
(FFT(trend="poly"), 11.4), | |||
(NaiveSeasonal(), 32.4), | |||
(KalmanForecaster(dim_x=3), 17.0), | |||
(LinearRegressionModel(lags=12), 11.0), | |||
(RandomForest(lags=12, n_estimators=200, max_depth=3), 15.5), |
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.
Isn't n_estimators=200
a bit too much for a unit test? This doesn't take too long?
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 just took this from a few lines below, tbh. I can reduce it.
@@ -361,6 +423,15 @@ def fit( | |||
**kwargs | |||
Additional keyword arguments passed to the `fit` method of the model. | |||
""" | |||
|
|||
self.encoders = self.initialize_encoders() | |||
if self.encoders.encoding_available: |
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.
Neat 👌
|
||
logger = get_logger(__name__) | ||
|
||
try: |
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.
Thanks!
Fixes #693.
Summary
add_encoders
(as already available for TorchForecastingModels)