gym-examples 2.0.31__py3-none-any.whl → 2.0.33__py3-none-any.whl

gym_examples/__init__.py

@@ -6,4 +6,4 @@ register(
      max_episode_steps=50,
 )
 
-__version__ = "2.0.31"
+__version__ = "2.0.33"

gym_examples/envs/wsn_env.py

@@ -115,9 +115,11 @@ class WSNRoutingEnv(gym.Env):
             # Calculate final reward
             # rewards_individual = torch.tensor(rewards[i], dtype=torch.double)                
             # final_reward = net(rewards_individual)
-            final_reward = np.sum(rewards[i])
+            final_reward = np.sum(rewards[i]) 
             rewards[i] = final_reward
 
+        rewards = [r * self.network_reward_consumption_energy() * self.network_reward_dispersion_remaining_energy() for r in rewards]
+
         for i in range(self.n_sensors):
             if (self.remaining_energy[i] <= 0) or (self.number_of_packets[i] <= 0):
                 dones[i] = True
@@ -254,6 +256,29 @@ class WSNRoutingEnv(gym.Env):
 
         return [reward_angle, reward_distance, reward_consumption_energy, reward_dispersion_remaining_energy, reward_number_of_packets]
 
+    def network_reward_dispersion_remaining_energy(self):
+        '''
+        Compute the reward based on the standard deviation of the remaining energy at the network level
+        '''
+        dispersion_remaining_energy = np.std(self.remaining_energy)
+        # Normalize the standard deviation of the remaining energy
+        max_dispersion_remaining_energy = initial_energy / 2 # maximum standard deviation of the remaining energy if n_sensors is even
+        normalized_dispersion_remaining_energy = dispersion_remaining_energy / max_dispersion_remaining_energy
+
+        return np.clip(1 - normalized_dispersion_remaining_energy, 0, 1)
+    
+    def network_reward_consumption_energy(self):
+        '''
+        Compute the reward based on the total energy consumption (transmission, reception) at the network level
+        '''
+        total_energy = np.sum(self.consumption_energy)
+        # Normalize the total energy consumption
+        max_transmission_energy = self.transmission_energy(self.n_sensors * initial_number_of_packets, self.coverage_radius)
+        max_reception_energy = self.reception_energy(self.n_sensors * initial_number_of_packets)
+        normalized_total_energy = total_energy / (max_transmission_energy + max_reception_energy)
+
+        return 1 - normalized_total_energy
+
     def integrate_mobility(self):
         '''
         Integrate the mobility of the sensors after each step

{gym_examples-2.0.31.dist-info → gym_examples-2.0.33.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gym-examples
-Version: 2.0.31
+Version: 2.0.33
 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-2.0.33.dist-info/RECORD

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=TbbQUrHAZ6jdbW4uDeI4-EHJTElY9I7xpSmyJ6EFScQ,193
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=xULVyW1wLeyt2xHmwEUuGsHUAL1lM-0zTAgnCM816QI,14726
+gym_examples-2.0.33.dist-info/METADATA,sha256=nLo7viKJw02WOBE1ItIEDTnn3RZLAK8fkOLywy5PfDs,411
+gym_examples-2.0.33.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-2.0.33.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-2.0.33.dist-info/RECORD,,

gym_examples-2.0.31.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=-Zi6pjtuGP4mWLhMJ5XoRMBap3wEBnvbILvg_aozuTw,193
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=M2bo_-wLWkXTJajQc7OgkT7vsXlCyGBTkfXA3ipFSc0,13265
-gym_examples-2.0.31.dist-info/METADATA,sha256=XZzdSuDCJ2CXoAF6osyaCT0nKN9xRQoL5GksbJvxnq4,411
-gym_examples-2.0.31.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-2.0.31.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-2.0.31.dist-info/RECORD,,