summaryrefslogtreecommitdiffstats
path: root/R Scripts/predict-victims-plots.R
diff options
context:
space:
mode:
authorBen Green <bgreen@g.harvard.edu>2015-09-18 23:38:46 -0400
committerBen Green <bgreen@g.harvard.edu>2015-09-18 23:38:48 -0400
commitbb68c49552f69b90ce840dbd7346fc76eb93e3e3 (patch)
treea5a184d2403eb16dbadf9cbf9a1f32f540d5ab00 /R Scripts/predict-victims-plots.R
parenta585b261dace47783fdae28130c8e4792627a230 (diff)
downloadcriminal_cascades-bb68c49552f69b90ce840dbd7346fc76eb93e3e3.tar.gz
pretty plots!
Diffstat (limited to 'R Scripts/predict-victims-plots.R')
-rw-r--r--R Scripts/predict-victims-plots.R8
1 files changed, 4 insertions, 4 deletions
diff --git a/R Scripts/predict-victims-plots.R b/R Scripts/predict-victims-plots.R
index 179a60b..0e1b669 100644
--- a/R Scripts/predict-victims-plots.R
+++ b/R Scripts/predict-victims-plots.R
@@ -24,7 +24,7 @@ counts = matrix(c( sum(correct_rank1<(vcount(lcc)*popsizes[1])),
sum(correct_rank3<(vcount(lcc)*popsizes[2])),
sum(correct_rank3<(vcount(lcc)*popsizes[3]))),
nrow=3, byrow=T)
-cols = brewer.pal(3,'Paired'); cols=cols[c(3,1,2)]
+cols = brewer.pal(3,'Dark2');# cols=cols[c(3,1,2)]
barplot(counts, border=NA,
xlab="Size of High-Risk Population",
ylab="Victims Predicted",
@@ -43,10 +43,10 @@ mtext(side = 4, line = 3, "Number of Victims Predicted")
#### Precision-Recall Curve
-plot(ecdf(correct_rank1),col=cols[1],lwd=2,main='',
+plot(ecdf(correct_rank1),col=cols[1],lwd=3,main='',
xlab='Ranking',ylab='CDF',xlim=c(0,140000))
-plot(ecdf(correct_rank2),col=cols[2],lwd=2,add=T)
-plot(ecdf(correct_rank3),col=cols[3],lwd=2,add=T)
+plot(ecdf(correct_rank2),col=cols[2],lwd=3,add=T)
+plot(ecdf(correct_rank3),col=cols[3],lwd=3,add=T)
legend("bottomright", inset=0.05,
c("Demographics Model", "Contagion Model", "Combined Model"),
fill=cols,border=NA,bty='n')