mednet.data.classify.typing

Define specialized data typing for classification tasks.

Classes

RawDataLoader()

A loader object can load samples and labels from storage for classification tasks.

class mednet.data.classify.typing.RawDataLoader[source]

Bases: RawDataLoader

A loader object can load samples and labels from storage for classification tasks.

target(sample)[source]

Load only sample target from its raw representation.

Parameters:

sample (Any) – Information about the sample to load. Implementation dependent.

Return type:

Tensor

Returns:

The label corresponding to the specified sample, encapsulated as a 1D torch float tensor.