matsengrp / torchdms

Analyze deep mutational scanning data with PyTorch
https://matsengrp.github.io/torchdms/
2 stars 0 forks source link

dropna-strata-ceiling #163

Closed jgallowa07 closed 2 years ago

jgallowa07 commented 2 years ago

Description

This PR:

  1. fixes a bug with "--drop-nan" functionality
  2. Adds "--strata-ceiling" to group strata above some threshold together during training.

Closes #(replace this parenthetical with issue number)

Tests

Please describe the tests added to verify correct behavior.

Checklist: