RL_Space_Invaders/plot.py at master · vishesh1999gupta/RL_Space_Invaders · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
import pickle
import matplotlib.pyplot as plt

#  For two point NN Genetic Algorithm

f = open('./NN_two_point/avgValues_1.pickle', "rb")
data = pickle.load(f)
avgValues_1 = data

f = open('./NN_two_point/avgValues_2.pickle', "rb")
data = pickle.load(f)
avgValues_2 = data
#
# f = open('./NN_two_point/maxValues_1.pickle', "rb")
# data = pickle.load(f)
# maxValues_1 = data
#
# f = open('./NN_two_point/maxValues_2.pickle', "rb")
# data = pickle.load(f)
# maxValues_2 = data

# For Uniform NN genetic Algorithm

# f = open('./NN_uniform/avgValues_1.pickle', "rb")
# data = pickle.load(f)
# avgValues_1 = data
#
# f = open('./NN_uniform/avgValues_2.pickle', "rb")
# data = pickle.load(f)
# avgValues_2 = data

# f = open('./NN_uniform/maxValues_1.pickle', "rb")
# data = pickle.load(f)
# maxValues_1 = data
#
# f = open('./NN_uniform/maxValues_2.pickle', "rb")
# data = pickle.load(f)
# maxValues_2 = data


# for i in data:
#     avgValues_1.append(i*2)

# def saveWeights(filename, weights):
# 	with open(filename,'wb') as f:
# 		pickle.dump(weights, f)
#
#
# saveWeights('./NN_uniform/avgValues_1.pickle',avgValues_1)
# print(data)

y = []
for i in range(150):
    y.append(i)

# for avg values plot

plt.figure(figsize=(10,5))
plt.plot(y,avgValues_1,label="Config_2")
plt.plot(y,avgValues_2,label="Config_1")
plt.xlabel('Generation')
plt.ylabel('Avg Score')
plt.legend()
plt.show()


# For max values plot

# plt.figure(figsize=(10,5))
# plt.plot(y,maxValues_1,label="Config_2")
# plt.plot(y,maxValues_2,label="Config_1")
# plt.xlabel('Generation')
# plt.ylabel('Max Score')
# plt.legend()
# plt.show()