From 7147679c428d732f36749d0ef37d2b13d364a655 Mon Sep 17 00:00:00 2001 From: mhz Date: Mon, 1 Jul 2024 10:04:51 +0200 Subject: [PATCH] EdgeMetricsCE only has 0 and 1 --- graph_dit/metrics/molecular_metrics_train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/graph_dit/metrics/molecular_metrics_train.py b/graph_dit/metrics/molecular_metrics_train.py index f141fb1..7200a9d 100644 --- 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):