Skip to content

Commit 0d720e1

Browse files
author
Alexander Ororbia
committed
minor tweak to sig-figs printing in probe utils
1 parent 9de3c98 commit 0d720e1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ngclearn/utils/analysis/probe.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -150,9 +150,9 @@ def fit(self, dataset, dev_dataset=None, n_iter=50, patience=20):
150150
L = (_L * x_mb.shape[0]) + L ## we remove the batch division from loss w.r.t. x_mb/y_mb
151151

152152
if dev_data is not None:
153-
print_string = f"\r{ii} L = {L / Ns:.3f} Acc = {acc / Ns:.2f} Dev.Acc = {best_acc:.2f}"
153+
print_string = f"\r{ii} L = {L / Ns:.4f} Acc = {acc / Ns:.4f} Dev.Acc = {best_acc:.4f}"
154154
else:
155-
print_string = f"\r{ii} L = {L / Ns:.3f} Acc = {acc / Ns:.2f}"
155+
print_string = f"\r{ii} L = {L / Ns:.4f} Acc = {acc / Ns:.4f}"
156156

157157
if hasattr(self, "eta"):
158158
print_string += f" LR = {getattr(self, 'eta'):.6f}"

0 commit comments

Comments
 (0)