EdgeMetricsCE only has 0 and 1

2024-07-01 10:04:51 +02:00
parent 817ef04c58
commit 7147679c42
1 changed files with 1 additions and 1 deletions
--- a/graph_dit/metrics/molecular_metrics_train.py
+++ b/graph_dit/metrics/molecular_metrics_train.py
@@ -85,7 +85,7 @@ class EdgeMetricsCE(MetricCollection):
        ce_SI = SingleCE(1)
        ce_DO = DoubleCE(2)
        ce_TR = TripleCE(3)
-        super().__init__([ce_no_bond, ce_SI, ce_DO, ce_TR])
+        super().__init__([ce_no_bond, ce_SI])

 class AtomMetricsCE(MetricCollection):
    def __init__(self, active_atoms):