normalise sasa values

pegerto · Jan 7, 2024 · 68b0b2e · 68b0b2e
1 parent 1ea470f
commit 68b0b2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/graphpro/annotations.py b/src/graphpro/annotations.py
@@ -76,4 +76,4 @@ def generate(self, G: Graph, atom_group: AtomGroup):
 
     def encode(self, G: Graph) -> torch.tensor:
         total_area = [G.node_attr(n)[self.attr_name] if self.attr_name in G.node_attr(n) else 0 for n in G.nodes()]
-        return torch.tensor([total_area], dtype=torch.float).T
+        return F.normalize(torch.tensor([total_area], dtype=torch.float).T)