summaryrefslogtreecommitdiff
path: root/scripts/pa-tex-table.rb
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/pa-tex-table.rb')
-rwxr-xr-xscripts/pa-tex-table.rb19
1 files changed, 13 insertions, 6 deletions
diff --git a/scripts/pa-tex-table.rb b/scripts/pa-tex-table.rb
index b163ab3..5d9821d 100755
--- a/scripts/pa-tex-table.rb
+++ b/scripts/pa-tex-table.rb
@@ -1,5 +1,6 @@
#!/usr/bin/env ruby
+#1 & 2 & 3 & 4 & 5 & 6 & 7 & 8 & 9 & Measured & MP2D & CDK & LR-sgd & LR-scikit & NN & RF \\\\
puts '
\definecolor{red}{rgb}{1,0,0}
\definecolor{darkred}{rgb}{0.5,0,0}
@@ -9,20 +10,25 @@ puts '
\definecolor{black}{rgb}{0,0,0}
\definecolor{white}{rgb}{1,1,1}
\tiny
-\begin{longtable}{rrrrrrrrrcccccccccc}
+\begin{longtable}{rrrrrrrrrccccccccccccccc}
\caption{Summary of pyrrolizidine alkaloid predictions: red: mutagen, green: non-mutagen, grey: no prediction, dark red/green: low confidence; 1: Retronecine, 2: Otonecine, 3: Platynecine, 4: N-oxide, 5: Dehydropyrrolizidine, 6:Tertiary PA, 7: Macrocyclic-diester, 8: Monoester, 9: Diester} \\\\
\label{tab:pa}
-1 & 2 & 3 & 4 & 5 & 6 & 7 & 8 & 9 & Measured & MP2D & CDK & DL & SVM & LR-sgd & LR-scikit & NN & RF \\kill % needed as guide for multicolumn
-\multicolumn{9}{c}{PA Group} & & \multicolumn{2}{c}{lazar} & \multicolumn{4}{c}{Tensorflow}\\\\
+% 1 & 2 & 3 & 4 & 5 & 6 & 7 & 8 & 9 & Measured & MP2D & CDK & DL & SVM & LR-sgd & LR-scikit & NN & RF \\kill % needed as guide for multicolumn
+% \multicolumn{9}{c}{PA Group} & & \multicolumn{2}{c}{lazar} & \multicolumn{4}{c}{Tensorflow}\\\\
+1 & 2 & 3 & 4 & 5 & 6 & 7 & 8 & 9 & Exp. & MP2D & CDK & LR1 & LR2 & NN & RF & SVM & LR1 & LR2 & NN & RF & SVM \\kill % needed as guide for multicolumn
+\multicolumn{9}{c}{PA Group} & & \multicolumn{2}{c}{lazar} & \multicolumn{5}{c}{MP2D} & \multicolumn{5}{c}{CDK}\\\\
-1 & 2 & 3 & 4 & 5 & 6 & 7 & 8 & 9 & Measured & MP2D & CDK & LR-sgd & LR-scikit & NN & RF \\\\
+1 & 2 & 3 & 4 & 5 & 6 & 7 & 8 & 9 & Exp. & MP2D & CDK & LR1 & LR2 & NN & RF & SVM & LR1 & LR2 & NN & RF & SVM \\\\
+% 1 & 2 & 3 & 4 & 5 & 6 & 7 & 8 & 9 & Measured & MP2D & CDK & LR-sgd & LR-scikit & NN & RF \\\\
\hline
\renewcommand{\arraystretch}{0.075}
'
File.read(ARGV[0]).each_line do |l|
unless l.match("SMILES")
- id,r,o,p,n,de,t,ma,mo,di,cid,smi,cansmi,exp,lazar_MP2D,lazar_MP2D_high_confidence,lazar_CDK,lazar_CDK_high_confidence,tf_lr_sgd,tf_lr_scikit,tf_NN,tf_RF = l.chomp.split(",")
+ #ID,Retronecine,Otonecine,Platynecine,N-oxide,Dehydropyrrolizidine,Tertiary PA,Macrocyclic-diester,Monoester,Diester,CID,SMILES,Canonical SMILES,Measured,lazar-MP2D,lazar-high-confidence-MP2D,lazar-CDK,lazar-high-confidence-CDK,LR-sgd-MP2D,LR-scikit-MP2D,NN-MP2D,RF-MP2D,SVM-MP2D,LR-sgd-CDK,LR-scikit-CDK,NN-CDK,RF-CDK,SVM-CDK
+ id,r,o,p,n,de,t,ma,mo,di,cid,smi,cansmi,exp,lazar_MP2D,lazar_MP2D_high_confidence,lazar_CDK,lazar_CDK_high_confidence,lr_sgd_mp2d,lr_scikit_mp2d,nn_mp2d,rf_mp2d,svm_mp2d,lr_sgd_cdk,lr_scikit_cdk,nn_cdk,rf_cdk,svm_cdk = l.chomp.split(",")
+ #id,r,o,p,n,de,t,ma,mo,di,cid,smi,cansmi,exp,lazar_MP2D,lazar_MP2D_high_confidence,lazar_CDK,lazar_CDK_high_confidence,tf_lr_sgd,tf_lr_scikit,tf_NN,tf_RF = l.chomp.split(",")
row = [r,o,p,n,de,t,ma,mo,di].collect{|group| group == "1" ? '\cellcolor{black}' : '\cellcolor{white}'}.join(' & ')
if exp == "1"
row += ' & \cellcolor{red}'
@@ -49,7 +55,8 @@ File.read(ARGV[0]).each_line do |l|
else
row += ' & \cellcolor{grey}'
end
- [tf_lr_sgd,tf_lr_scikit,tf_NN,tf_RF].each do |mut|
+ [lr_sgd_mp2d,lr_scikit_mp2d,nn_mp2d,rf_mp2d,svm_mp2d,lr_sgd_cdk,lr_scikit_cdk,nn_cdk,rf_cdk,svm_cdk].each do |mut|
+ #[tf_lr_sgd,tf_lr_scikit,tf_NN,tf_RF].each do |mut|
mut == "1" ? row += ' & \cellcolor{red}' : row += ' & \cellcolor{green}'
end
puts row + ' \\\\'