Mercurial > repos > iuc > virhunter

diff predict.py @ 2:ea2cccb9f73e draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/galaxyproject/tools-iuc/tree/master/tools/VirHunter commit c3685ed6a70b47012b62b95a2a3db062bd3b7475
author iuc
date Thu, 05 Jan 2023 14:27:54 +0000
parents 9b12bc1b1e2c
children 302332b914ef
line wrap: on
line diff
--- a/predict.py	Wed Nov 30 17:31:52 2022 +0000
+++ b/predict.py	Thu Jan 05 14:27:54 2023 +0000
@@ -104,7 +104,7 @@
         df.groupby(["id", "length", 'RF_decision'], sort=False).size().unstack(fill_value=0)
     )
     df = df.reset_index()
-    df = df.reindex(['length', 'id', 'virus', 'plant', 'bacteria'], axis=1)
+    df = df.reindex(['length', 'id', 'virus', 'plant', 'bacteria'], axis=1).fillna(value=0)
     conditions = [
         (df['virus'] > df['plant']) & (df['virus'] > df['bacteria']),
         (df['plant'] > df['virus']) & (df['plant'] > df['bacteria']),