Mercurial > repos > jay > pdaug_word_vector_model
view PDAUG_Peptide_Data_Access/PDAUG_Peptide_Data_Access.py @ 9:984238f609d4 draft
"planemo upload for repository https://github.com/jaidevjoshi83/pdaug commit d396d7ff89705cc0dd626ed32c45a9f4029b1b05"
author | jay |
---|---|
date | Mon, 10 Jan 2022 04:49:15 +0000 |
parents | 682fee9fef16 |
children |
line wrap: on
line source
import modlamp from modlamp.datasets import load_AMPvsTM from modlamp.datasets import load_AMPvsUniProt from modlamp.datasets import load_ACPvsTM from modlamp.datasets import load_ACPvsRandom from modlamp.database import query_apd from modlamp.database import query_camp import os import pandas as pd def DataGen(DataBaseType, OutFile, IDs): if DataBaseType == 'AMPvsTMP': data = load_AMPvsTM() elif DataBaseType == 'AMPvsUniProt': data = load_AMPvsUniProt() elif DataBaseType == 'ACPvsTMP': data = load_ACPvsTM() elif DataBaseType == 'ACPvsRandom': data = load_ACPvsRandom() elif DataBaseType == 'query_apd': data = query_apd([int(i) for i in IDs.split(',')]) df = pd.DataFrame(data, columns=['Peptides']) df.to_csv(OutFile, index=False, sep='\t') exit() elif DataBaseType == 'query_camp': data = query_camp([int(i) for i in IDs.split(',')]) df = pd.DataFrame(data, columns=['Peptides']) df.to_csv(OutFile, index=False, sep='\t') exit() else: print ("Enter Correct Values") exit() peptide_data = data.sequences class_label = int(len(peptide_data)/2)*[data.target_names[0]]+int(len(peptide_data)/2)*[data.target_names[1]] peptide_data = pd.DataFrame(peptide_data, columns=['name']) class_label = pd.DataFrame(class_label, columns=['class_label']) df = pd.concat([peptide_data,class_label], axis=1) df.to_csv(OutFile, index=False, sep='\t') if __name__=="__main__": import argparse parser = argparse.ArgumentParser() parser.add_argument("-d", "--DataBaseType", required=True, default=None, help="Name of the dataset ") parser.add_argument("-o", "--OutFile", required=False, default='Out.tsv', help="Out put file name for str descriptors") parser.add_argument("-L", "--List", required=False, default=None, help="List of integer as ID") args = parser.parse_args() DataGen(args.DataBaseType, args.OutFile, args.List)