oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove redundant cols from transactions

Browse Source
This commit is contained in:
Marek Wolan
2023-06-30 10:41:56 +01:00
parent c3c4512544
commit 2a8d28cba6
4 changed files with 8 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/primaite/transactions/transaction.py
View File

@@ -20,23 +20,14 @@ class Transaction(object):
self.episode_number = _episode_number
self.step_number = _step_number
def set_obs_space_pre(self, _obs_space_pre):
def set_obs_space(self, _obs_space):
"""
Sets the observation space (pre).
Args:
_obs_space_pre: The observation space before any actions are taken
"""
self.obs_space_pre = _obs_space_pre
def set_obs_space_post(self, _obs_space_post):
"""
Sets the observation space (post).
Args:
_obs_space_post: The observation space after any actions are taken
"""
self.obs_space_post = _obs_space_post
self.obs_space = _obs_space
def set_reward(self, _reward):
"""

9
src/primaite/transactions/transactions_to_file.py
View File

@@ -58,12 +58,12 @@ def write_transaction_to_file(
action_header.append("AS_" + str(x))
# Create the observation space headers array
obs_header_initial = [f"pre_{o}" for o in obs_space_description]
obs_header_new = [f"post_{o}" for o in obs_space_description]
# obs_header_initial = [f"pre_{o}" for o in obs_space_description]
# obs_header_new = [f"post_{o}" for o in obs_space_description]
# Open up a csv file
header = ["Timestamp", "Episode", "Step", "Reward"]
header = header + action_header + obs_header_initial + obs_header_new
header = header + action_header + obs_space_description
try:
filename = session_path / f"all_transactions_{timestamp_str}.csv"
@@ -82,8 +82,7 @@ def write_transaction_to_file(
csv_data = (
csv_data
+ turn_action_space_to_array(transaction.action_space)
+ transaction.obs_space_pre.tolist()
+ transaction.obs_space_post.tolist()
+ transaction.obs_space.tolist()
)
csv_writer.writerow(csv_data)