Merge pull request #10 from PumasAI-Labs/htbunn-patch-1

Update cts_distributed.jl

Author: andreasnoack

simulation/qualification.jl

@@ -132,7 +132,7 @@ coreqpc = map(zip(inspectdfs, coreparams)) do (idf, param)
         plt = plot_qpc(refq50, q5, q50, q95, refvalue, "50th Percentile for "*uppercase(string(p)))
 
         try
-            save(joinpath("results",OUTDIR,string(param)*"_qpc.png"), plt)
+            save(joinpath("results",OUTDIR,string(p)*"_qpc.png"), plt)
         catch e
             @info "Could not save $(string(p)) QPC plot"
         end
@@ -183,9 +183,9 @@ derivedqpc = map(zip(inspectdfs, observedvals)) do (idf, val)
         plt = plot_qpc(refq50, q5, q50, q95, refvalue, "50th Percentile for "*uppercase(string(v)))
 
         try
-            save(joinpath("results",OUTDIR,string(val)*"_qpc.png"), plt)
+            save(joinpath("results",OUTDIR,string(v)*"_qpc.png"), plt)
         catch e
-            @info "Could not save $(string(val)) QPC plot"
+            @info "Could not save $(string(v)) QPC plot"
         end
 
         return plt
@@ -228,4 +228,4 @@ try
     save(joinpath("results",OUTDIR,"sdmaredbl_qpc.png"), plt)
 catch e
     @info "Could not save SDMA reduction QPC plot"
-end
+end

simulation/template/cts_distributed.jl

@@ -64,7 +64,7 @@ ADJ6MG = (
 ##################################################################
 @info "Load Simulation Functions"
 # load simulation functions
-include("./utils/sim_functions.jl")
+@everywhere include("./utils/sim_functions.jl")
 
 
 ##################################################################
@@ -74,6 +74,9 @@ include("./utils/sim_functions.jl")
 # Set number of subjects to simulate in each scenario
 subjects_to_sim = 500;
 
+# Set number of cycles
+ncycles = 3;
+
 # empty dataframe to sim info and results
 mysims = DataFrame();
 
@@ -114,7 +117,7 @@ end;
 mysims[!, :sims] = pmap(eachrow(mysims)) do r
     map(eachrow(r.patients)) do subr
         try
-            return sim_trial(subr, 3);
+            return sim_trial(subr, ncycles);
         catch e
             return e
         end
@@ -126,5 +129,5 @@ end;
 ##################################################################
 @info "Serialize Results"
 open(result_filename, "w") do io
-    serialize(io, sims_df)
-end
+    serialize(io, mysims)
+end