0.5.7 predictions in output are sorted by score

senseppi/__init__.py

-__version__ = "0.5.6"
+__version__ = "0.5.7"
 __author__ = "Konstantin Volzhenin"
 
 from . import model, commands, esm2_model, dataset, utils, network_utils

senseppi/commands/predict.py

@@ -149,9 +149,11 @@ def main(params):
                          'protein name 1, protein name 2 and label(optional)')
     data['preds'] = preds
 
+    data = data.sort_values(by=['preds'], ascending=False)
     data.to_csv(params.output + '.tsv', sep='\t', index=False, header=True)
 
     data_positive = data[data['preds'] >= params.pred_threshold]
+    data_positive = data_positive.sort_values(by=['preds'], ascending=False)
     data_positive.to_csv(params.output + '_positive_interactions.tsv', sep='\t', index=False, header=True)
 
     if os.path.isfile(tmp_pairs):