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

Add the validation score and training time for create_function in XGBoost #1327

Merged
merged 4 commits into from
Nov 2, 2023

Conversation

jineetd
Copy link
Contributor

@jineetd jineetd commented Oct 31, 2023

Let us show the validation score and training time for the XGBoost AutoML model trained. This shall give us fair enough idea on how the model trained on the training data set.

@@ -674,7 +682,10 @@ def exec(self, *args, **kwargs):
msg = f"Function {self.node.name} overwritten."
else:
msg = f"Function {self.node.name} added to the database."
yield Batch(pd.DataFrame([msg]))
if best_score and train_time:
yield Batch(pd.DataFrame([msg, best_score, train_time]))
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Could we add some text? Other wise, it will be just two numbers without any explanation.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Added some annotations

@xzdandy
Copy link
Collaborator

xzdandy commented Nov 1, 2023

@xzdandy xzdandy merged commit f409057 into georgia-tech-db:staging Nov 2, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants