zenithsiz-ist/arc-proj
1
0
Fork 0
mirror of https://github.com/Zenithsiz/arc-proj.git synced 2026-02-03 14:09:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed satisfaction color bar not taking into account the threshold.

Browse Source
This commit is contained in:
Filipe Rodrigues 2023-10-24 10:17:40 +01:00
parent 6309b5fd4c
commit 33db83ef14
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/main.py
View File

@ -156,7 +156,7 @@ def run(params: RunParams):
ax.imshow(graph.satisfaction_img())
# Add the color bar for the satisfaction
colors = [(1, 0, 0), (0, 0, 0), (0, 1, 0)]
colors = [(0.0, (1, 0, 0)), (graph.agent_ty.threshold(), (0, 0, 0)), (1.0, (0, 1, 0))]
fig.colorbar(
mpl.cm.ScalarMappable(
norm=mpl.colors.Normalize(vmin=0.0, vmax=1.0),