myModel Class Reference

Inheritance diagram for myModel:

Public Member Functions
def	__init__ (self, number_of_features)
def	forward (self, x)

Public Attributes
	network
	a dense model with one hidden layer

Detailed Description

My dense neural network

Definition at line 45 of file torch.py.

Constructor & Destructor Documentation

__init__()

def __init__	(		self,
			number_of_features
	)

Init the network
param: number_of_features number of input variables

Definition at line 50 of file torch.py.

    def __init__(self, number_of_features):
        """
        Init the network
        param: number_of_features number of input variables
        """
        super(myModel, self).__init__()
 
        
        self.network = torch.nn.Sequential(
            torch.nn.Linear(number_of_features, 128),
            torch.nn.ReLU(),
            torch.nn.Linear(128, 128),
            torch.nn.ReLU(),
            torch.nn.Linear(128, 1),
            torch.nn.Sigmoid(),
        )
 

Member Function Documentation

forward()

def forward	(		self,
			x
	)

Run the network

Definition at line 67 of file torch.py.

    def forward(self, x):
        """
        Run the network
        """
        prob = self.network(x)
        return prob
 
 

Member Data Documentation

network

network

a dense model with one hidden layer

Definition at line 58 of file torch.py.

---

The documentation for this class was generated from the following file:

• mva/scripts/basf2_mva_python_interface/torch.py