Skip to content

Commit 8673057

Browse files
committed
improve compatibility
1 parent 06070f1 commit 8673057

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

model.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,8 @@
5353
# visualize testing
5454
ax1.cla()
5555
for c in range(4):
56-
bp = ax1.bar(x=c+0.1, height=sum((np.argmax(pred_, axis=1) == c)), width=0.2, color='red')
57-
bt = ax1.bar(x=c-0.1, height=sum((np.argmax(test_data[:, 21:], axis=1) == c)), width=0.2, color='blue')
56+
bp = ax1.bar(c+0.1, height=sum((np.argmax(pred_, axis=1) == c)), width=0.2, color='red')
57+
bt = ax1.bar(c-0.1, height=sum((np.argmax(test_data[:, 21:], axis=1) == c)), width=0.2, color='blue')
5858
ax1.set_xticks(range(4), ["accepted", "good", "unaccepted", "very good"])
5959
ax1.legend(handles=[bp, bt], labels=["prediction", "target"])
6060
ax1.set_ylim((0, 400))

0 commit comments

Comments
 (0)