Mercurial > repos > recetox > matchms_networking
annotate matchms_networking_wrapper.py @ 0:e4ec3592507f draft
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
author | recetox |
---|---|
date | Tue, 18 Oct 2022 13:23:28 +0000 |
parents | |
children | 8147d93d372d |
rev | line source |
---|---|
0
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
1 import argparse |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
2 import sys |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
3 |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
4 from matchms.importing import scores_from_json |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
5 from matchms.networking import SimilarityNetwork |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
6 |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
7 |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
8 def main(argv): |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
9 parser = argparse.ArgumentParser(description="Create network-graph from similarity scores.") |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
10 parser.add_argument("--graph_format", type=str, help="Format of the output similarity network.") |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
11 parser.add_argument("--identifier", type=str, help="Unique metadata identifier of each spectrum from which scores are computed.") |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
12 parser.add_argument("--top_n", type=int, help="Number of highest-score edges to keep.") |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
13 parser.add_argument("--max_links", type=int, help="Maximum number of links to add per node.") |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
14 parser.add_argument("--score_cutoff", type=float, help="Minimum similarity score value to link two spectra.") |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
15 parser.add_argument("--link_method", type=str, help="Method for selecting top N edges for each node.") |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
16 parser.add_argument("--keep_unconnected_nodes", help="Keep unconnected nodes in the network.", action="store_true") |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
17 parser.add_argument("scores", type=str, help="Path to matchMS similarity-scores .json file.") |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
18 parser.add_argument("output_filename", type=str, help="Path where to store the output similarity network.") |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
19 args = parser.parse_args() |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
20 |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
21 scores = scores_from_json(args.scores) |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
22 |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
23 network = SimilarityNetwork(identifier_key=args.identifier, |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
24 top_n=args.top_n, |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
25 max_links=args.max_links, |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
26 score_cutoff=args.score_cutoff, |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
27 link_method=args.link_method, |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
28 keep_unconnected_nodes=args.keep_unconnected_nodes) |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
29 |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
30 network.create_network(scores) |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
31 network.export_to_file(filename=args.output_filename, graph_format=args.graph_format) |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
32 |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
33 return 0 |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
34 |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
35 |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
36 if __name__ == "__main__": |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
37 main(argv=sys.argv[1:]) |
e4ec3592507f
planemo upload for repository https://github.com/RECETOX/galaxytools/tree/master/tools/matchms commit f7bab98744e338dcdbdc9cf6f9de287632c76ea2
recetox
parents:
diff
changeset
|
38 pass |