optimas-org / optimas

Optimization at scale, powered by libEnsemble
https://optimas.readthedocs.io
Other
22 stars 13 forks source link

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

Closed AngelFP closed 3 months ago

AngelFP commented 7 months ago

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