doc
Define models used in documentation tutorials.
Classes:
Name | Description |
---|---|
EcgCNN |
Ecg model, backbone model for ECG dataset. |
GazPriceMLP |
Gas Price model, task_learner model for gas price dataset. |
MnistCNN |
MNIST model, backbone model for MNIST dataset. |
EcgCNN(dropout: float = 0.0, output_size: int = 50, *, out_channels: tuple[int, ...] = (32, 64, 128), with_softmax: bool = False)
#
Ecg model, backbone model for ECG dataset.
Initialize a basic CNN model for classification.
Methods:
Name | Description |
---|---|
forward |
Forward pass. |
Source code in src/xpdeep/model/zoo/doc.py
GazPriceMLP(input_size: int, hidden_channels: list[int], norm_layer: Callable[..., torch.nn.Module] | None = None, activation_layer: Callable[..., torch.nn.Module] | None = torch.nn.ReLU, dropout: float = 0.0, last_activation: partial[torch.nn.Module] | None = None, *, inplace: bool | None = None, bias: bool = True, flatten_input: bool = False)
#
Gas Price model, task_learner model for gas price dataset.
Methods:
Name | Description |
---|---|
forward |
Forward pass of the model. |