galaxy.models.cnn_mlp

Classes

Functions

load_model(in_channels[, n_out_classes])

Module Contents

class galaxy.models.cnn_mlp.CNN_MLP(in_channels: int, n_out_classes: int = 1)

Bases: torch.nn.Module

cnn_layers

mlp_layers

forward(x)

galaxy.models.cnn_mlp.load_model(in_channels: int, n_out_classes: int = 1)