CASBI.utils.CNN

Module Contents

Classes

ConvNet_halo	This class defines the CNN architecture for the embedding network. The CNN is used to extract the features from the chemical abundance histograms of entire galaxy halos.
ConvNet_subhalo	This class defines the CNN architecture for the embedding network. The CNN is used to extract the features from the chemical abundance histograms of single dwarf galaxy subhalos.

API

class CASBI.utils.CNN.ConvNet_halo(output_dim)

Bases: torch.nn.Module

This class defines the CNN architecture for the embedding network. The CNN is used to extract the features from the chemical abundance histograms of entire galaxy halos.

Initialization

Initialize internal Module state, shared by both nn.Module and ScriptModule.

set_device(device)

forward(x)

class CASBI.utils.CNN.ConvNet_subhalo(output_dim)

Bases: torch.nn.Module

This class defines the CNN architecture for the embedding network. The CNN is used to extract the features from the chemical abundance histograms of single dwarf galaxy subhalos.

Initialization

Initialize internal Module state, shared by both nn.Module and ScriptModule.

set_device(device)

forward(x)