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.py
contains the training algorithm.