gym-examples 3.0.334__py3-none-any.whl → 3.0.336__py3-none-any.whl

gym_examples/__init__.py

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

gym_examples/envs/wsn_env.py

@@ -54,7 +54,11 @@ class WSNRoutingEnv(gym.Env):
     def __init__(self, n_sensors = 20, coverage_radius=(upper_bound - lower_bound)/4, num_timesteps = None, version = None):
 
         super(WSNRoutingEnv, self).__init__()
-
+        # Create filenames to save statistics for evaluation
+        self.statistics_filename = "../results/data/Statistics_filename_" + os.getenv('ALGO_NAME') + ".txt"
+        with open(self.statistics_filename, 'w') as file:
+            file.write("Episode return, Total of energy consumption, Std of remaining energy, Mean of remaining energy, \
+                       Average latency, Network throughput, Packet delivery ratio, Network lifetime\n")
         # Initialize list of episode metrics
         self.num_timesteps = num_timesteps # This argument is for the PPO algorithm
         self.version = version # This argument is for the PPO algorithm
@@ -86,11 +90,29 @@ class WSNRoutingEnv(gym.Env):
         self.reset()        
 
     def reset(self):
+        if self.number_of_steps > 30000: # Change 30000 to a parameter taken from args or kwargs
+            self.episode_returns.append(self.episode_return)
+            observations = self._get_obs()
+            remaining_energy_values = np.array([sensor['remaining_energy'] for sensor in observations])
+            std_remaining_energy = np.std(remaining_energy_values)
+            mean_remaining_energy = np.mean(remaining_energy_values)
+            consumption_energy_values = np.array([sensor['consumption_energy'] for sensor in observations])
+            total_consumption_energy = np.sum(consumption_energy_values)
+
+            # Append the statistics to the .txt file
+            with open(self.statistics_filename, 'a') as file:
+                file.write(f"{self.episode_return}, {total_consumption_energy}, {std_remaining_energy}, {mean_remaining_energy}\
+                           {self.average_latency}, {self.network_throughput}, {self.packet_delivery_ratio}, {self.network_lifetime}\n")
 
-        if self.number_of_steps > 3: # Change 30000 to a parameter taken from args or kwargs
             print("Episode count: ", self.episode_count)
-            print("get_metrics: ", self.get_metrics())
-            print(f"self._get_obs(): {self._get_obs()}")
+            print("Episode return: ", self.episode_return)
+            print("Total consumption energy: ", total_consumption_energy)
+            print("Std remaining energy: ", std_remaining_energy)
+            print("Mean remaining energy: ", mean_remaining_energy)            
+            print("Average latency: ", self.average_latency)
+            print("Network throughput: ", self.network_throughput)
+            print("Packet delivery ratio: ", self.packet_delivery_ratio)
+            print("Network lifetime: ", self.network_lifetime)
 
         self.episode_return = 0              
         self.sensor_positions = np.random.rand(self.n_sensors, 2) * (upper_bound - lower_bound) + lower_bound
@@ -206,6 +228,7 @@ class WSNRoutingEnv(gym.Env):
                 dones[i] = self.remaining_energy[i] <= 0 or self.number_of_packets[i] == 0
         dones = bool(np.all(dones))
 
+        self.episode_return += rewards
         return self._get_obs(), rewards, dones, self.get_metrics()
 
 

{gym_examples-3.0.334.dist-info → gym_examples-3.0.336.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=a2Ws3UTCWQzuNccW3gwQPx82jqZpDGK-4QIRyVobyX4,166
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=fPTN2K9Wo6N8TiL9b9CCF4MSrdyv4_h5i68haPagIkI,25556
+gym_examples-3.0.336.dist-info/METADATA,sha256=b2Ib3GLoq0oMrEjqD_5EAjeD6nU5x7-YdR1MFQulHCw,412
+gym_examples-3.0.336.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-3.0.336.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-3.0.336.dist-info/RECORD,,

gym_examples-3.0.334.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=icrDtusfsDMkeXkb9cxtrM1qmjLBTIRXAd7bUUup2Fk,166
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=2bRJvedMHD4Amjk_hRvtcoaXZ4Gv_Fp7sZvhg74_KIo,23701
-gym_examples-3.0.334.dist-info/METADATA,sha256=MYSkA-LbZZGL6mnGAMXZTU4C3UuqHvjXZmIr7Z_ooWI,412
-gym_examples-3.0.334.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-3.0.334.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-3.0.334.dist-info/RECORD,,