0.3.1 small bugfix for predict_string

parent 93ff261b
__version__ = "0.3.0" __version__ = "0.3.1"
__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
......
...@@ -29,6 +29,7 @@ def main(params): ...@@ -29,6 +29,7 @@ def main(params):
generate_pairs_string(fasta_file, output_file=pairs_file, delete_proteins=params.delete_proteins) generate_pairs_string(fasta_file, output_file=pairs_file, delete_proteins=params.delete_proteins)
params.fasta_file = fasta_file params.fasta_file = fasta_file
params.pairs_file = pairs_file
compute_embeddings(params) compute_embeddings(params)
preds = predict(params) preds = predict(params)
...@@ -193,7 +194,7 @@ def add_args(parser): ...@@ -193,7 +194,7 @@ def add_args(parser):
string_pred_args.add_argument("-n", "--nodes", type=int, default=10, string_pred_args.add_argument("-n", "--nodes", type=int, default=10,
help="Number of nodes to fetch from STRING database. Default: 10") help="Number of nodes to fetch from STRING database. Default: 10")
string_pred_args.add_argument("-r", "--score", type=int, default=0, string_pred_args.add_argument("-r", "--score", type=int, default=0,
help="Score threshold for STRING connections. Range: (0, 1000). Default: 500") help="Score threshold for STRING connections. Range: (0, 1000). Default: 0")
string_pred_args.add_argument("-p", "--pred_threshold", type=int, default=500, string_pred_args.add_argument("-p", "--pred_threshold", type=int, default=500,
help="Prediction threshold. Range: (0, 1000). Default: 500") help="Prediction threshold. Range: (0, 1000). Default: 500")
string_pred_args.add_argument("--graphs", action='store_true', string_pred_args.add_argument("--graphs", action='store_true',
......
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