fix normalizefeatures transform for negative values

pyg-team · Jan 5, 2022 · 6008e30 · 6008e30
1 parent 696da2c
commit 6008e30
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torch_geometric/transforms/normalize_features.py b/torch_geometric/transforms/normalize_features.py
@@ -17,7 +17,9 @@ def __init__(self, attrs: List[str] = ["x"]):
     def __call__(self, data: Union[Data, HeteroData]):
         for store in data.stores:
             for key, value in store.items(*self.attrs):
-                store[key] = value / value.sum(dim=-1, keepdim=True).clamp_(1.)
+                value = value - value.min()
+                value.div_(value.sum(dim=-1, keepdim=True).clamp_(min=1.))
+                store[key] = value
         return data
 
     def __repr__(self) -> str: