modulation.classification.procedure.single_epoch_evaluation

modulation.classification.procedure.single_epoch_evaluation(model: Module, evaluation_loader: DataLoader, subset: SplitEnum, *, epoch: Optional[int] = None, writer: Optional[Writer] = None, device: device = device(type='cuda')) → float[source]
Parameters
  • model –

  • evaluation_loader –

  • subset –

  • epoch –

  • writer –

  • device –

Returns

Return type