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

Expose AxModelManager methods in Service API generators and add new plotting methods #169

Merged
merged 40 commits into from
May 23, 2024

Conversation

AngelFP
Copy link
Member

@AngelFP AngelFP commented Jan 27, 2024

This PR adds two new plotting methods to AxModelManager (plot_cross_validation and plot_feature_importance) and makes the AxServiceGenerators (i.e., all Ax generators based on the Service API) inherit from AxModelManager. This exposes all plotting methods and other utilities of the manager directly at the generator, which is convenient for interactive optimization.

plot_contour:
image
plot_cross_validation:
image
plot_slice:
image
plot_feature_importance:
image

@AngelFP AngelFP added the enhancement New feature or request label Jan 27, 2024
AngelFP and others added 11 commits January 28, 2024 02:35
The previous implementation was failing for numpy dtypes, as it was assuming the use of native python types. This prevented the generator
from being initalized, for example, from an `ExplorationDiagnostics`. This is because the parameters in the diags are read from the serialized json file, and thus always initialized with the numpy dtype name.
@delaossa delaossa assigned delaossa and unassigned delaossa Feb 14, 2024
@delaossa delaossa requested review from delaossa and removed request for delaossa February 14, 2024 11:43
@delaossa delaossa assigned delaossa and unassigned delaossa Feb 14, 2024
@AngelFP AngelFP changed the title [WIP] Expose Ax plotting utilities in Service API generators [WIP] Expose AxModelManager methods in Service API generators and add new plotting methods Mar 7, 2024
@AngelFP AngelFP changed the title [WIP] Expose AxModelManager methods in Service API generators and add new plotting methods Expose AxModelManager methods in Service API generators and add new plotting methods Mar 7, 2024
@AngelFP AngelFP requested a review from delaossa March 8, 2024 11:04
Copy link
Collaborator

@delaossa delaossa left a comment

Choose a reason for hiding this comment

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

I have been doing some tests and it seems to be working as expected.
Very useful to check how the model evolves in interactive sessions!
Thanks Angel!

@AngelFP AngelFP merged commit 742fc71 into main May 23, 2024
6 checks passed
@AngelFP AngelFP deleted the feature/expose_ax_plots branch May 23, 2024 12:06
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants