Browse Source

node

tags/v0.3.1
lihy96 Frozenmad 4 years ago
parent
commit
b271df41c3
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      autogl/module/train/node_classification_full.py

+ 1
- 1
autogl/module/train/node_classification_full.py View File

@@ -229,7 +229,7 @@ class NodeClassificationFullTrainer(BaseNodeClassificationTrainer):
if hasattr(data, "val_mask") and data.val_mask is not None:
if type(self.feval) is list:
feval = self.feval[0]
else:
else:branch
feval = self.feval
val_loss = self.evaluate([data], mask=data.val_mask, feval=feval)
if feval.is_higher_better() is True:


Loading…
Cancel
Save