diff predict.py @ 3:302332b914ef draft default tip

planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/VirHunter commit 58587e05f604590c70550e13fc51b7425e916ed4
author iuc
date Sat, 14 Jan 2023 21:08:33 +0000
parents ea2cccb9f73e
children
line wrap: on
line diff
--- a/predict.py	Thu Jan 05 14:27:54 2023 +0000
+++ b/predict.py	Sat Jan 14 21:08:33 2023 +0000
@@ -153,7 +153,6 @@
                 length=l_,
                 use_10=use_10
             )
-            print(df)
             df = predict_rf(
                 df=df,
                 rf_weights_path=weights,
@@ -168,14 +167,14 @@
         df_500 = dfs_fr[0][(dfs_fr[0]['length'] >= limit) & (dfs_fr[0]['length'] < 1500)]
         df_1000 = dfs_fr[1][(dfs_fr[1]['length'] >= 1500)]
         df = pd.concat([df_1000, df_500], ignore_index=True)
-        pred_fr = Path(out_path, 'predicted_fragments.csv')
-        df.to_csv(pred_fr)
+        pred_fr = Path(out_path, 'predicted_fragments.tsv')
+        df.to_csv(pred_fr, sep='\t')
 
         df_500 = dfs_cont[0][(dfs_cont[0]['length'] >= limit) & (dfs_cont[0]['length'] < 1500)]
         df_1000 = dfs_cont[1][(dfs_cont[1]['length'] >= 1500)]
         df = pd.concat([df_1000, df_500], ignore_index=True)
-        pred_contigs = Path(out_path, 'predicted.csv')
-        df.to_csv(pred_contigs)
+        pred_contigs = Path(out_path, 'predicted.tsv')
+        df.to_csv(pred_contigs, sep='\t')
 
         if return_viral:
             viral_ids = list(df[df["decision"] == "virus"]["id"])