pub_soft/foldingdiff
3
0
Fork 0
mirror of https://github.com/microsoft/foldingdiff.git synced 2026-06-04 13:30:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Report lengths of angle vectors

Browse Source
This commit is contained in:
Kevin Wu
2022-07-05 21:44:48 +00:00
parent 9a7a3035d2
commit 1fb95639f3
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
protdiff/datasets.py
View File

@@ -61,12 +61,19 @@ class CathConsecutiveAnglesDataset(Dataset):
pool.join()
for s, a in zip(self.structures, angles):
s["angles"] = a
# Remove items with nan in angles/structures
orig_count = len(self.structures)
self.structures = [s for s in self.structures if s["angles"] is not None]
new_count = len(self.structures)
logging.info(f"Removed structures with nan {orig_count} -> {new_count}")
# Aggregate the lengths
all_lengths = [s["angles"].shape[1] for s in self.structures]
logging.info(
f"Length of angles: {np.min(all_lengths)}-{np.max(all_lengths)}, mean {np.mean(all_lengths)}"
)
def __len__(self) -> int:
"""Returns the length of this object"""
return len(self.structures)