diff options
Diffstat (limited to 'scripts/pa-tex-table.rb')
-rwxr-xr-x | scripts/pa-tex-table.rb | 19 |
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 + ' \\\\' |