1 files changed, 17 insertions, 6 deletions
diff --git a/scripts/pa-table.rb b/scripts/pa-table.rb
index 032aaa1..d808fa1 100755
--- a/scripts/pa-table.rb
+++ b/scripts/pa-table.rb
@@ -7,10 +7,10 @@ puts '
 \definecolor{lightgreen}{rgb}{0,0.5,0}
 \definecolor{grey}{rgb}{0.5,0.5,0.5}
 \tiny
-\begin{longtable}{rcccccc}
+\begin{longtable}{rcccccccccc}
 \caption{Summary of pyrrolizidine alkaloid predictions: red: mutagen, green: non-mutagen, grey: no prediction, dark red/green: low confidence} \\\\
 \label{tab:pa}
- & & \multicolumn{2}{c}{lazar} & \multicolumn{3}{c}{R} \\\\
+ & & \multicolumn{2}{c}{lazar} & \multicolumn{3}{c}{R} & \multicolumn{4}{c}{Tensorflow}\\\\
 '
 #header = ["ID","measured","lazar"]
 header = ["ID","Measured","MP2D","PaDEL"]
@@ -59,8 +59,8 @@ File.read("pyrrolizidine-alkaloids/lazar/pa-padel-predictions.csv").each_line do
   end
   i += 1
 end
-Dir["pyrrolizidine-alkaloids/PA.*.outcome.csv"].each do |r|
-  header << r.sub('pyrrolizidine-alkaloids/PA.','').sub('.outcome.csv','')
+Dir["pyrrolizidine-alkaloids/R/PA.*.outcome.csv"].each do |r|
+  header << r.sub('pyrrolizidine-alkaloids/R/PA.','').sub('.outcome.csv','')
   i = 0
   File.read(r).each_line do |l|
     if i > 0
@@ -75,13 +75,24 @@ Dir["pyrrolizidine-alkaloids/PA.*.outcome.csv"].each do |r|
     i += 1
   end
 end
+Dir["pyrrolizidine-alkaloids/tensorflow/pred.*.v3-ext-Padel-2D.csv"].each do |r|
+  header << r.sub('pyrrolizidine-alkaloids/tensorflow/pred.','').sub('.v3-ext-Padel-2D.csv','').sub("lr2","LR-scikit").sub("lr","LR-sgd").sub("rf","RF").sub("nn","NN")
+  i = 0
+  File.read(r).each_line do |l|
+    if i > 0
+      id,pred = l.chomp.split(",")
+      pred.to_f > 0.5 ? tab[i-1] << ' & \cellcolor{red}' : tab[i-1] << ' & \cellcolor{green}'
+    end
+    i += 1
+  end
+end
 #tab.collect!{|t| t + '\cellcolor{grey}  \\\\'}
 puts header.join(" & ")+" \\\\"
 puts '\hline'
 puts '\renewcommand{\arraystretch}{0.075}'
 tab.collect!{|t| t + ''}
-puts tab.join(" \\\\ \n")
-puts '
+print tab.join(" \\\\ \n")
+puts ' \\\\
 \end{longtable}
 \normalsize
 '