braceal / molecules

Machine learning for molecular dynamics.
MIT License
5 stars 5 forks source link

t-SNE callback #21

Closed braceal closed 4 years ago

braceal commented 4 years ago
  1. Take random sample of train and validation contact matrices.
  2. Compute RMSD to native state and fraction of native contacts for sampled matrices. 2.1 Use openmm RMSD and fraction of native contacts reporters.
  3. Make a callback which stores a matplotlib plot as a member variable. The plot has the axis/labels and color map formatted at initialization of the callback. The only thing that changes when the callback is called during training is the coordinates of the t-SNE embeddings. Then during training we can observe the points moving. Draw train samples as circles and validation samples as triangles.
  4. Save dictionary of embeddings and indices to disk, save plot png to disk.
  5. Add images to tensorboard during training.

Bonus: Make a movie of the embeddings in bokeh

braceal commented 4 years ago

Done