已测试
该算法现阶段与tensorboard的scalar smoothing
操作匹配 (时间:2022-7-23)
代码网址: https://github.com/ustcml/mini-lightning/blob/main/mini_lightning/_visualize.py
import matplotlib.pyplot as plt
import numpy as np
from typing import Tuple
from matplotlib.axes import Axes
from matplotlib.figure import Figure
from typing import List, Dict
from tensorboard.backend.event_processing.event_accumulator import EventAccumulator
def config_ax(ax: Axes, title: str = None, xlabel: str = None, ylabel: str = None,
xlim: Tuple[float, float] = None, ylim: Tuple[float, float] = None,
xticks: List[float] = None, xticks_labels: List[str] = None,
yticks: List[float] = None, yticks_labels: List[str