Module ml4opf.viz.table

Functions

def make_stats_df(data: dict[str, torch.Tensor], round_to: int | None = 5)

Browse git

Make a pandas DataFrame with mean/max/min/std stats.

Args

data : dict[str, Tensor]

Each tensor should be of shape [batch, 1]

round_to : Optional[int]

Number of decimals to round each Tensor to (default: 5)

Returns

DataFrame

mean/max/min/std along batch dimension.

When used with DataFrame.to_latex(), the output can be copied directly into a LaTeX table.