V14

Model definition not used in any publication

class doc_octopy.models.definitions.raul_net.online.v14.RaulNetV14(learning_rate, nr_of_input_channels, input_length__samples, nr_of_outputs, cnn_encoder_channels, mlp_encoder_channels, event_search_kernel_length, event_search_kernel_stride)

Model definition not used in any publication

Parameters:

• learning_rate (float)

• nr_of_input_channels (int)

• input_length__samples (int)

• nr_of_outputs (int)

• cnn_encoder_channels (Tuple[int, int, int])

• mlp_encoder_channels (Tuple[int, int])

• event_search_kernel_length (int)

• event_search_kernel_stride (int)

learning_rate

The learning rate.

Type:

float

nr_of_input_channels

The number of input channels.

Type:

int

nr_of_outputs

The number of outputs.

Type:

int

cnn_encoder_channels

Tuple containing 3 integers defining the cnn encoder channels.

Type:

Tuple[int, int, int]

mlp_encoder_channels

Tuple containing 2 integers defining the mlp encoder channels.

Type:

Tuple[int, int]

event_search_kernel_length

Integer that sets the length of the kernels searching for action potentials.

Type:

int

event_search_kernel_stride

Integer that sets the stride of the kernels searching for action potentials.

Type:

int

compute_loss(prediction, ground_truths)

forward(inputs)

Same as torch.nn.Module.forward.

Parameters:

• *args – Whatever you decide to pass into the forward method.

• **kwargs – Keyword arguments are also possible.

Returns:

Your model’s output

Return type:

Tensor