File size: 1,517 Bytes
8896a5f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
"""
D-SCRIPT: Structure Aware PPI Prediction
"""
import argparse, os, sys
class CitationAction(argparse.Action):
def __init__(self, option_strings, dest, **kwargs):
super(CitationAction, self).__init__(option_strings, dest, **kwargs)
def __call__(self, parser, namespace, values, option_string=None):
import dscript
print(dscript.__citation__)
setattr(namespace, self.dest, values)
sys.exit(0)
def main():
import dscript
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument("-v", "--version", action="version", version="D-SCRIPT " + dscript.__version__)
parser.add_argument(
"-c",
"--citation",
action=CitationAction,
nargs=0,
help="show program's citation and exit",
)
subparsers = parser.add_subparsers(title="D-SCRIPT Commands", dest="cmd")
subparsers.required = True
import dscript.commands.train
import dscript.commands.eval
import dscript.commands.embed
import dscript.commands.predict
modules = {
"train": dscript.commands.train,
"eval": dscript.commands.eval,
"embed": dscript.commands.embed,
"predict": dscript.commands.predict,
}
for name, module in modules.items():
sp = subparsers.add_parser(name, description=module.__doc__)
module.add_args(sp)
sp.set_defaults(func=module.main)
args = parser.parse_args()
args.func(args)
if __name__ == "__main__":
main()
|