ljvmiranda921 / gym-lattice

An HP 2D Lattice Environment with a Gym-like API for the Protein Folding Problem

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Refactor observations

ljvmiranda921 opened this issue · comments

I think it's better for our observation space to be a grid rather than a chain:

  • Create a matrix of shape (m, m) matrix where m = 2 * len(seq) + 1
  • Get the midpoint of that grid and treat that as origin. That's why you want to have odd number of cells
  • The grid should have values in {-1, 0, 1} for polar, unassigned, and hydrophobic separately