s452638/msc-patryk-bartkowiak
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed evaluate function

Browse Source
This commit is contained in:
Patryk Bartkowiak 2024-10-21 20:12:20 +00:00
parent 240c16b495
commit 3f116b583d
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
code/src/train_codebert_mlm.py
View File

@ -208,6 +208,7 @@ def evaluate(model: PreTrainedModel, dataloader: DataLoader) -> Tuple[float, flo
avg_loss: float = total_loss / len(dataloader) avg_loss: float = total_loss / len(dataloader)
avg_acc: float = total_acc / len(dataloader.dataset) avg_acc: float = total_acc / len(dataloader.dataset)
model.train()
return avg_loss, avg_acc return avg_loss, avg_acc
def main() -> None: def main() -> None: