Update SimTools.jl

jiewenTsai · Dec 25, 2024 · 9af60f5 · 9af60f5
1 parent cbb4939
commit 9af60f5
Showing 1 changed file with 24 additions and 8 deletions.
diff --git a/src/SimTools.jl b/src/SimTools.jl
@@ -498,10 +498,18 @@ end
 """
 """
 function getMetrics(object; par=:a)
-    vectorTrue = object[:True][Symbol(par)]
-    vectorEsti = Array{Any}(undef, length(vectorTrue), 100)
-    for i in 1:100
-        vectorEsti[:,i] = object[i][Symbol(par)]
+    if par == (:β)
+        vectorTrue = object[:True][Symbol(par)]
+        vectorEsti = Array{Any}(undef, length(vectorTrue), 100)
+        for i in 1:100
+            vectorEsti[:,i] = object[i][Symbol(par)][2:end]
+        end 
+    else
+        vectorTrue = object[:True][Symbol(par)]
+        vectorEsti = Array{Any}(undef, length(vectorTrue), 100)
+        for i in 1:100
+            vectorEsti[:,i] = object[i][Symbol(par)]
+        end
     end
 
     Metrics = (
@@ -518,10 +526,18 @@ end
 """
 """
 function getMetrics2(object; par=:a)
-    vectorTrue = object[:True][Symbol(par)]
-    vectorEsti = Array{Any}(undef, length(vectorTrue), 100)
-    for i in 1:100
-        vectorEsti[:,i] = object[i][Symbol(par)]
+    if par == (:β)
+        vectorTrue = object[:True][Symbol(par)]
+        vectorEsti = Array{Any}(undef, length(vectorTrue), 100)
+        for i in 1:100
+            vectorEsti[:,i] = object[i][Symbol(par)][2:end]
+        end 
+    else
+        vectorTrue = object[:True][Symbol(par)]
+        vectorEsti = Array{Any}(undef, length(vectorTrue), 100)
+        for i in 1:100
+            vectorEsti[:,i] = object[i][Symbol(par)]
+        end
     end
 
     Metrics = (