-
Notifications
You must be signed in to change notification settings - Fork 28
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
Xgboost #405
base: main
Are you sure you want to change the base?
Conversation
Signed-off-by: Kushal Batra <[email protected]>
Signed-off-by: Kushal Batra <[email protected]>
Signed-off-by: Kushal Batra <[email protected]>
Codecov ReportAttention: Patch coverage is
Additional details and impacted files@@ Coverage Diff @@
## main #405 +/- ##
==========================================
- Coverage 92.11% 92.02% -0.10%
==========================================
Files 97 99 +2
Lines 4782 4900 +118
Branches 430 442 +12
==========================================
+ Hits 4405 4509 +104
- Misses 279 289 +10
- Partials 98 102 +4 ☔ View full report in Codecov by Sentry. |
Signed-off-by: Kushal Batra <[email protected]>
Signed-off-by: Kushal Batra <[email protected]>
Signed-off-by: Kushal Batra <[email protected]>
Signed-off-by: Kushal Batra <[email protected]>
Signed-off-by: Kushal Batra <[email protected]>
Signed-off-by: Kushal Batra <[email protected]>
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.
Few comments
|
||
def _check_data_format(df) -> bool: | ||
if not isinstance(df, pd.DataFrame): | ||
raise TypeError("df should be of type pd.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.
Can you cover these lines with tests?
def predict_horizon(self, df: pd.DataFrame) -> np.ndarray: | ||
_check_data_format(df) | ||
transformed_test_data = CovariatesGenerator().transform(df) | ||
_LOGGER.info("Predicting the horizon") |
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.
remove log
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.
or maybe use debug
pyproject.toml
Outdated
@@ -43,6 +43,8 @@ redis = { extras = ["hiredis"], version = "^5.0", optional = true } | |||
boto3 = { version = "^1.24.64", optional = true } | |||
pydruid = { version = "^0.6", optional = true } | |||
PyMySQL = { version = "^1.1.0", optional = true } | |||
xgboost = "^2.1.0" | |||
darts = "^0.30.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.
change this to use a lighter version. Also add a TODO to remove this
Note: We will have to revisit on how to optimize imports for xgboost and darts