TS2ReadoutSpec

class ibl_bwb_eval.TS2ReadoutSpec(id, mask_dim, primary_metric, metrics)

Bases: object

What one TS2 task predicts.

id

the task name.

mask_dim

the dim of a (batch, time, units) tensor the held-out mask varies along. Co-smoothing holds out units, forecasting holds out timesteps; the mask is constant along every other dim.

primary_metric

the metric used for model selection and for best/*/avg.

metrics

mapping from metric name to a callable that instantiates the metric.