forked from keras-team/keras-io
-
Notifications
You must be signed in to change notification settings - Fork 1
/
KID_comparison_graph.py
15 lines (12 loc) · 1.07 KB
/
KID_comparison_graph.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
import matplotlib.pyplot as plt
sigmoid_kid = [2.0346, 1.8818, 1.6469, 1.3623, 1.2319, 1.0647, 0.9277, 0.8640, 0.8051, 0.6772, 0.5296, 0.4812, 0.4360, 0.4272, 0.4095, 0.4066, 0.4126, 0.4183, 0.4202, 0.3996, 0.4235, 0.4089, 0.4119, 0.3972, 0.3839, 0.3801, 0.3747, 0.3599, 0.4232, 0.4060, 0.3856, 0.3718, 0.3646, 0.3618, 0.3896, 0.4050, 0.3924, 0.3771, 0.4217, 0.4646, 0.4712, 0.4850, 0.4686, 0.4409, 0.4032, 0.3805, 0.3655, 0.3983, 0.4068, 0.4114]
prelu_kid = [2.0456, 1.9287, 1.6846, 1.5458, 1.8669, 2.6438, 1.9348, 1.3044, 1.0837, 0.9490, 0.8596, 0.7928, 0.7505, 0.6936, 0.6649, 0.5972, 0.5554, 0.5093, 0.4807, 0.4620, 0.4439, 0.4326, 0.4288, 0.4231, 0.4260, 0.4230, 0.4234, 0.4227, 0.4158, 0.4008, 0.3935, 0.3805, 0.3778, 0.3746, 0.3718, 0.3638, 0.3544, 0.3505, 0.3507, 0.3530, 0.3430, 0.3424, 0.3499, 0.3345, 0.3233, 0.3242, 0.3205, 0.3210, 0.3246, 0.3163]
# Plot KID values
plt.figure(figsize=(10, 6))
plt.plot(sigmoid_kid, label='Sigmoid', color='blue')
plt.plot(prelu_kid, label='PReLU', color='red')
plt.xlabel('Epochs')
plt.ylabel('KID')
plt.legend(fontsize=18)
plt.grid(True)
plt.show()