Model Class Reference

Inheritance diagram for Model:

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 28 of file simple.py.

Constructor & Destructor Documentation

__init__()

__init__	(		self,
			number_of_features )

Parameters:
  number_of_features: number of input features

Definition at line 33 of file simple.py.

    def __init__(self, number_of_features):
        """
        Parameters:
          number_of_features: number of input features
        """
        super().__init__()
 
        
        self.network = nn.Sequential(
            nn.Linear(number_of_features, 128),
            nn.ReLU(),
            nn.Linear(128, 128),
            nn.ReLU(),
            nn.Linear(128, 1),
            nn.Sigmoid(),
        )
 

Member Function Documentation

forward()

forward	(		self,
			x )

Run the network

Definition at line 50 of file simple.py.

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

Member Data Documentation

network

network

Initial value:

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

a dense model with one hidden layer

Definition at line 41 of file simple.py.

---

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

• mva/examples/onnx/simple.py