gym-examples 3.0.369__py3-none-any.whl → 3.0.371__py3-none-any.whl

gym_examples/__init__.py

@@ -5,4 +5,4 @@ register(
      entry_point="gym_examples.envs:WSNRoutingEnv",
 )
 
-__version__ = "3.0.369"
+__version__ = "3.0.371"

gym_examples/envs/wsn_env.py

@@ -36,7 +36,8 @@ class Attention(nn.Module):
         self.output_dim = output_dim  # Set the output dimension of the network
         self.linear1 = nn.Linear(input_dim, 64)  # Define the first linear layer. It takes input of size 'input_dim' and outputs size '64'
         self.linear2 = nn.Linear(64, output_dim)  # Define the second linear layer. It takes input of size '64' and outputs size 'output_dim'
-
+        self.softplus = nn.Softplus()  # SoftPlus activation function to ensure non-negative outputs
+        
     def forward(self, x):
         # Step 1: Ensure input is 2D by adding a batch dimension if necessary
         if x.dim() == 1:
@@ -45,6 +46,7 @@ class Attention(nn.Module):
         attention_weights = F.softmax(x, dim=1)  # Apply the softmax function to get the attention weights. Initially F.softmax(x, dim=0) 
         x = attention_weights * x  # Multiply the input by the attention weights
         x = self.linear2(x)  # Pass the result through another linear layer
+        x = self.softplus(x)  # Apply SoftPlus activation to ensure non-negative outputs
         return x
 
 net = Attention(input_dim, output_dim)

{gym_examples-3.0.369.dist-info → gym_examples-3.0.371.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gym-examples
-Version: 3.0.369
+Version: 3.0.371
 Summary: A custom environment for multi-agent reinforcement learning focused on WSN routing.
 Home-page: https://github.com/gedji/CODES.git
 Author: Georges Djimefo

gym_examples-3.0.371.dist-info/RECORD

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=xgcRtA4LxuxW_KTDA4A4EH1H_1G6m6_PWlTj5ENOeBY,166
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=2NUSYwbm1Ygaff5T7qzehDZp7S-DYMYPcvXF6sQahFA,26116
+gym_examples-3.0.371.dist-info/METADATA,sha256=CA5-mEjTRxY_g6XQcX-bPmIgnN_xx6E-THwdYKbkg2M,412
+gym_examples-3.0.371.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-3.0.371.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-3.0.371.dist-info/RECORD,,

gym_examples-3.0.369.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=Sh5NvlSXNWnNESr6Jj3uZm2FSMyBdy_7hRdSLVZpSSk,166
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=erPf4tvrmD-M4ISTP9wHNTnVGgjZQYLqBWrVutIKcw8,25916
-gym_examples-3.0.369.dist-info/METADATA,sha256=pLMzNuC4BxjKO1PzZkndxSZj8vZVVkot6Y573-zBhes,412
-gym_examples-3.0.369.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-3.0.369.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-3.0.369.dist-info/RECORD,,