Mercurial > repos > iuc > virhunter
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"])