Skip to content

S
SENSE-PPI
Commit c963949c by Konstantin Volzhenin
Options

0.5.7 predictions in output are sorted by score

parent a183543b
Showing with 3 additions and 1 deletions
senseppi/__init__.py
__version__ = "0.5.6" __version__ = "0.5.7"
__author__ = "Konstantin Volzhenin" __author__ = "Konstantin Volzhenin"
from . import model, commands, esm2_model, dataset, utils, network_utils from . import model, commands, esm2_model, dataset, utils, network_utils
......
senseppi/commands/predict.py
...@@ -149,9 +149,11 @@ def main(params): ...@@ -149,9 +149,11 @@ def main(params):
'protein name 1, protein name 2 and label(optional)') 'protein name 1, protein name 2 and label(optional)')
data['preds'] = preds data['preds'] = preds
data = data.sort_values(by=['preds'], ascending=False)
data.to_csv(params.output + '.tsv', sep='\t', index=False, header=True) data.to_csv(params.output + '.tsv', sep='\t', index=False, header=True)
data_positive = data[data['preds'] >= params.pred_threshold] 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) data_positive.to_csv(params.output + '_positive_interactions.tsv', sep='\t', index=False, header=True)
if os.path.isfile(tmp_pairs): if os.path.isfile(tmp_pairs):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment