File size: 885 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 |
"""
Generate new embeddings using pre-trained language model.
"""
import argparse
from dscript.language_model import embed_from_fasta
def add_args(parser):
"""
Create parser for command line utility.
:meta private:
"""
parser.add_argument("--seqs", help="Sequences to be embedded", required=True)
parser.add_argument("-o", "--outfile", help="h5 file to write results", required=True)
parser.add_argument("-d", "--device", type=int, default=-1, help="Compute device to use")
return parser
def main(args):
"""
Run embedding from arguments.
:meta private:
"""
inPath = args.seqs
outPath = args.outfile
device = args.device
embed_from_fasta(inPath, outPath, device, verbose=True)
if __name__ == "__main__":
parser = argparse.ArgumentParser(description=__doc__)
add_args(parser)
main(parser.parse_args())
|