Skip to content

Conversation

Innixma
Copy link
Contributor

@Innixma Innixma commented Feb 18, 2025

Issue #, if available:

Description of changes:

Adding extra features to better support TabRepo 2.0:

  • Add predictor.model_hyperparameters(model)
  • Add predictor.model_info(model)
  • Add include_feature_metadata boolean to model.get_info(), as feature_metadata is dependent on AutoGluon unlike all other outputs of get_info which are python native types.

TODO:

  • Add unit test for holdout
  • Add unit test for bagging
  • (Stretch) Add predictor.export_hyperparameters(models) to return a usable hyperparameters dict that can be passed into a new predictor.fit(... hyperparameters=hyperparameters) call to reproduce a given AutoGluon stack architecture on new tasks.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@Innixma Innixma added enhancement New feature or request module: tabular labels Feb 18, 2025
@Innixma Innixma added this to the 1.3 Release milestone Feb 18, 2025
Copy link

Job PR-4901-d9e9761 is done.
Docs are uploaded to http://autogluon-staging.s3-website-us-west-2.amazonaws.com/PR-4901/d9e9761/index.html

@@ -2243,6 +2255,7 @@ def get_info(self) -> dict:
"predict_time": self.predict_time,
"val_score": self.val_score,
"hyperparameters": self.params,
"hyperparameters_user": self.get_hyperparameters_init(),
Copy link
Contributor

Choose a reason for hiding this comment

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

Why is this needed ?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

This contains the exact hyperparameters specified by the user for the model. This way we can retrieve what the user specified the hyperparameters to be, which is different from what the hyperparameters actually are (since the hyperparameters of the model include the default hyperparameters for those unspecified by the user)

@Innixma Innixma merged commit f189494 into autogluon:master Feb 19, 2025
4 checks passed
Copy link

Job PR-4901-2acbaca is done.
Docs are uploaded to http://autogluon-staging.s3-website-us-west-2.amazonaws.com/PR-4901/2acbaca/index.html

@Innixma Innixma deleted the add_hyperparameters_user branch April 16, 2025 21:23
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request module: tabular
Projects
Status: Done
Development

Successfully merging this pull request may close these issues.

2 participants