aepsych icon indicating copy to clipboard operation
aepsych copied to clipboard

Implement a new version of PairwiseGPModel to work with the Ax backend

Open crasanders opened this issue 2 years ago • 0 comments

Currently PairwiseGPModel only works with the old backend, so it cannot take advantage of Ax features. We would like a new pairwise model that works with the Ax backend, while keeping the current model for backwards compatibility.

This new model would take in pairwise data in the same format as the current model, and it would support all of the same Ax features as the other Ax-enabled models including:

  • Discrete parameters
  • Fixed parameters
  • Log-scaled parameters
  • Integer parameters
  • Parameter constraints

ContinuousRegressionGP and BinaryClassificationGP can be used as templates for how to convert old-style models (GPRegressionModel and GPClassificationModel) into Ax-enabled models. A new example config should be written and one or more unit tests should be added to test_ax_integration.py to verify that the model works as expected.

crasanders avatar Dec 14 '22 17:12 crasanders