Pytorch based reimplementation of COMS: Conservative Objective Models for Effective Offline Model-Based Optimization.
The original tensorflow implementation is here.
run_coms.py is the entry point of the code.
pytorch_nets.py contains the surrogate model.
pytorch_coms.pycontains the training algorithm.