myModel Class Reference

Inheritance diagram for myModel:

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

Public Attributes
	network
	a dense model with one hidden layer

Detailed Description

My dense neural network

Definition at line 75 of file torch.py.

Constructor & Destructor Documentation

__init__()

__init__	(		self,
			number_of_features )

Init the network
param: number_of_features number of input variables

Definition at line 80 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 = pytorch.nn.Sequential(
            pytorch.nn.Linear(number_of_features, 128),
            pytorch.nn.ReLU(),
            pytorch.nn.Linear(128, 128),
            pytorch.nn.ReLU(),
            pytorch.nn.Linear(128, 1),
            pytorch.nn.Sigmoid(),
        )
 

Member Function Documentation

forward()

forward	(		self,
			x )

Run the network

Definition at line 97 of file torch.py.

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

Member Data Documentation

network

network

Initial value:

=  pytorch.nn.Sequential(
            pytorch.nn.Linear(number_of_features, 128),
            pytorch.nn.ReLU(),
            pytorch.nn.Linear(128, 128),
            pytorch.nn.ReLU(),
            pytorch.nn.Linear(128, 1),
            pytorch.nn.Sigmoid(),
        )

a dense model with one hidden layer

Definition at line 88 of file torch.py.

---

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

• mva/scripts/basf2_mva_python_interface/torch.py