iicd/Graph-DiT
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

EdgeMetricsCE only has 0 and 1

Browse Source
This commit is contained in:
mhz 2024-07-01 10:04:51 +02:00
parent 817ef04c58
commit 7147679c42
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
graph_dit/metrics/molecular_metrics_train.py
View File

@ -85,7 +85,7 @@ class EdgeMetricsCE(MetricCollection):
ce_SI = SingleCE(1) ce_SI = SingleCE(1)
ce_DO = DoubleCE(2) ce_DO = DoubleCE(2)
ce_TR = TripleCE(3) 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): class AtomMetricsCE(MetricCollection):
def __init__(self, active_atoms): def __init__(self, active_atoms):