The RobotNetwork class. More...
Inheritance diagram for rl.dqn.RobotNetwork:
Public Member Functions | |
| def | __init__ (self, int hidden_nodes, int number_of_actions, int state_size) |
| The RobotNetwork class initializer. More... | |
| def | forward (self, network_input) |
| Internal function of the network, used for processing the input of the network. More... | |
Public Attributes | |
| linear1 | |
| linear2 | |
| linear3 | |
Detailed Description
The RobotNetwork class.
Defines the class that will represent the network that is used for deep Q-learning.
Constructor & Destructor Documentation
◆ __init__()
| def rl.dqn.RobotNetwork.__init__ | ( | self, | |
| int | hidden_nodes, | ||
| int | number_of_actions, | ||
| int | state_size | ||
| ) |
The RobotNetwork class initializer.
- Parameters
-
hidden_nodes The number of hidden nodes for the hidden layers. number_of_actions The number of possible actions, important for nodes needed as output. state_size The number of variables in a state, important for the nodes needed as input.
- Returns
- An instance of the RobotNetwork class.
Member Function Documentation
◆ forward()
| def rl.dqn.RobotNetwork.forward | ( | self, | |
| network_input | |||
| ) |
Internal function of the network, used for processing the input of the network.
- Parameters
-
network_input The network input.
The documentation for this class was generated from the following file:
- src/rl/dqn.py
