Skip to content

Commit

Permalink
Resolve minor formatting issue
Browse files Browse the repository at this point in the history
  • Loading branch information
tomaarsen committed Sep 19, 2024
1 parent 35755c6 commit 64efc66
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/setfit/notebook.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ class SetFitNotebookProgressCallback(NotebookProgressCallback):
In particular, it accepts "embedding_loss", "aspect_embedding_loss", and "polarity_embedding_loss"
and the corresponding metrics for the validation set.
"""

def on_log(self, *args, logs=None, **kwargs):
if logs is not None:
logs = {key if key != "embedding_loss" else "loss": value for key, value in logs.items()}
Expand Down

0 comments on commit 64efc66

Please sign in to comment.