spaCy/spacy/training/loggers.py

102 lines
3.7 KiB
Python
Raw Normal View History

from typing import TYPE_CHECKING, Dict, Any, Tuple, Callable, List, Optional, IO
2020-10-03 17:31:58 +03:00
from wasabi import Printer
import tqdm
import sys
from ..util import registry
from ..errors import Errors
2020-10-03 17:31:58 +03:00
if TYPE_CHECKING:
from ..language import Language # noqa: F401
2020-10-11 13:55:46 +03:00
def setup_table(
*, cols: List[str], widths: List[int], max_width: int = 13
) -> Tuple[List[str], List[int], List[str]]:
final_cols = []
final_widths = []
for col, width in zip(cols, widths):
if len(col) > max_width:
col = col[: max_width - 3] + "..." # shorten column if too long
final_cols.append(col.upper())
final_widths.append(max(len(col), width))
return final_cols, final_widths, ["r" for _ in final_widths]
@registry.loggers("spacy.ConsoleLogger.v1")
2020-10-03 17:31:58 +03:00
def console_logger(progress_bar: bool = False):
def setup_printer(
2020-10-03 17:31:58 +03:00
nlp: "Language", stdout: IO = sys.stdout, stderr: IO = sys.stderr
) -> Tuple[Callable[[Optional[Dict[str, Any]]], None], Callable[[], None]]:
write = lambda text: print(text, file=stdout, flush=True)
2020-10-03 17:31:58 +03:00
msg = Printer(no_print=True)
2020-10-05 18:43:42 +03:00
# ensure that only trainable components are logged
logged_pipes = [
name
for name, proc in nlp.pipeline
if hasattr(proc, "is_trainable") and proc.is_trainable
2020-10-05 18:43:42 +03:00
]
eval_frequency = nlp.config["training"]["eval_frequency"]
2020-09-24 12:04:35 +03:00
score_weights = nlp.config["training"]["score_weights"]
score_cols = [col for col, value in score_weights.items() if value is not None]
loss_cols = [f"Loss {pipe}" for pipe in logged_pipes]
2020-10-11 13:55:46 +03:00
spacing = 2
table_header, table_widths, table_aligns = setup_table(
cols=["E", "#"] + loss_cols + score_cols + ["Score"],
widths=[3, 6] + [8 for _ in loss_cols] + [6 for _ in score_cols] + [6],
)
write(msg.row(table_header, widths=table_widths, spacing=spacing))
write(msg.row(["-" * width for width in table_widths], spacing=spacing))
progress = None
2020-10-03 17:31:58 +03:00
def log_step(info: Optional[Dict[str, Any]]) -> None:
nonlocal progress
if info is None:
# If we don't have a new checkpoint, just return.
if progress is not None:
progress.update(1)
2020-10-03 17:31:58 +03:00
return
losses = [
"{0:.2f}".format(float(info["losses"][pipe_name]))
2020-10-05 18:43:42 +03:00
for pipe_name in logged_pipes
]
2020-09-13 18:39:31 +03:00
scores = []
for col in score_cols:
2020-09-24 12:04:35 +03:00
score = info["other_scores"].get(col, 0.0)
try:
score = float(score)
except TypeError:
err = Errors.E916.format(name=col, score_type=type(score))
raise ValueError(err) from None
if col != "speed":
score *= 100
scores.append("{0:.2f}".format(score))
data = (
[info["epoch"], info["step"]]
+ losses
+ scores
+ ["{0:.2f}".format(float(info["score"]))]
)
if progress is not None:
progress.close()
2020-10-11 13:55:46 +03:00
write(
msg.row(data, widths=table_widths, aligns=table_aligns, spacing=spacing)
)
if progress_bar:
# Set disable=None, so that it disables on non-TTY
progress = tqdm.tqdm(
2020-10-03 17:31:58 +03:00
total=eval_frequency, disable=None, leave=False, file=stderr
)
progress.set_description(f"Epoch {info['epoch']+1}")
2020-10-03 17:31:58 +03:00
def finalize() -> None:
pass
return log_step, finalize
return setup_printer