Commit 85fd5e58 authored by Alexander Palmisano's avatar Alexander Palmisano
Browse files

added plots

parent 34fd4fca
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
......@@ -86,10 +86,10 @@ def read_data(file, shield_num, columns):
return pd.DataFrame(data_shield_num, columns=columns)
columns = ['episode', 'steps', 'mean reward', 'mean reward 1000', 'mean length', 'mean loss', 'time delta', 'shield']
data1 = read_data('./Agent/plots/data/map9_log.txt', 0, columns)
data2 = read_data('./Agent/plots/data/map9_shield1_log.txt', 1, columns)
data3 = read_data('./Agent/plots/data/map9_shield2_log.txt', 2, columns)
data4 = read_data('./Agent/plots/data/map9_shield3_log.txt', 3, columns)
data1 = read_data('./Agent/plots/data/map9_apples_log.txt', 0, columns)
data2 = read_data('./Agent/plots/data/map9_apples_shield1_log.txt', 1, columns)
data3 = read_data('./Agent/plots/data/map9_apples_shield2_log.txt', 2, columns)
data4 = read_data('./Agent/plots/data/map9_apples_shield3_log.txt', 3, columns)
data = pd.concat([data1, data2, data3, data4], ignore_index=True)
data = data.drop(data[data['steps']>270000].index)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment