upgraded plot_likelihood

ilabcode · Dec 25, 2023 · dfaa280 · dfaa280
1 parent ac83e76
commit dfaa280
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/functions.jl b/src/functions.jl
@@ -67,7 +67,8 @@ end
 """Function for Plotting Likelihood"""
 function plot_likelihood(A)
     # Create the heatmap
-    heatmap(A, color=:greys, clim=(0, 1),aspect_ratio = :equal, yflip=true, legend=false,xticks=1:9, yticks=1:9, xlabel="STATES", ylabel="OBSERVATIONS" )
+    x_max, y_max = size(A)
+    heatmap(A, color=:greys, clim=(0, 1),aspect_ratio = :equal, yflip=true, legend=false,xticks=1:x_max, yticks=1:y_max, xlabel="STATES", ylabel="OBSERVATIONS" )
 end