gym-examples 3.0.354__py3-none-any.whl → 3.0.355__py3-none-any.whl

gym_examples/__init__.py

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

gym_examples/envs/wsn_env.py

@@ -60,7 +60,7 @@ class WSNRoutingEnv(gym.Env):
         # 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, Network lifetime\n")
+            file.write("Episode return, Total of energy consumption, Std of remaining energy, Mean of remaining energy, Network lifetime, network_throughput, energy_efficiency, packet_delivery_ratio, network_lifetime, average_latency\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
@@ -95,6 +95,12 @@ class WSNRoutingEnv(gym.Env):
         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()
+            metrics = self.get_metrics()
+            network_throughput = metrics["network_throughput"]
+            energy_efficiency = metrics["energy_efficiency"]
+            packet_delivery_ratio = metrics["packet_delivery_ratio"]
+            network_lifetime = metrics["network_lifetime"]
+            average_latency = metrics["average_latency"]
             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)
@@ -103,7 +109,8 @@ class WSNRoutingEnv(gym.Env):
 
             # 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.network_lifetime}\n")
+                file.write(f"{self.episode_return}, {total_consumption_energy}, {std_remaining_energy}, {mean_remaining_energy}, {self.network_lifetime}, 
+                           {network_throughput}, {energy_efficiency}, {packet_delivery_ratio}, {network_lifetime}, {average_latency}\n")
 
             # print("Episode count: ", self.episode_count)
             # print("Episode return: ", self.episode_return)

{gym_examples-3.0.354.dist-info → gym_examples-3.0.355.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+gym_examples/__init__.py,sha256=lYLIhK75Cbobycsj0iM1GJQx4oTfWqlMb_hA_O3GlKU,166
+gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
+gym_examples/envs/wsn_env.py,sha256=Zujo-Ei4zNuz-8zf0TIPS2vHGDdxXXlWTZanb34ttIo,25945
+gym_examples-3.0.355.dist-info/METADATA,sha256=QQH20cVu7Hs_61WM64tki2qJ68iUQtX8A6vnm4z1wNU,412
+gym_examples-3.0.355.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
+gym_examples-3.0.355.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
+gym_examples-3.0.355.dist-info/RECORD,,

gym_examples-3.0.354.dist-info/RECORD

@@ -1,7 +0,0 @@
-gym_examples/__init__.py,sha256=HVlncGWSmQlSYKkKcDellZ2Eqs0VkaLm62-6zuttvus,166
-gym_examples/envs/__init__.py,sha256=lgMe4pyOuUTgTBUddM0iwMlETsYTwFShny6ifm8PGM8,53
-gym_examples/envs/wsn_env.py,sha256=Dv9LjdTJHfZ3c4NSIFNq9DePV4hFOkJ7IM9eCAkCpBE,25356
-gym_examples-3.0.354.dist-info/METADATA,sha256=a3JJX0Ly-Tj9MCFe6xgWRLKqLSCeuDkUG-9jqoDtkso,412
-gym_examples-3.0.354.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-gym_examples-3.0.354.dist-info/top_level.txt,sha256=rJRksoAF32M6lTLBEwYzRdo4PgtejceaNnnZ3HeY_Rk,13
-gym_examples-3.0.354.dist-info/RECORD,,