-
Notifications
You must be signed in to change notification settings - Fork 4
/
extractFeature.py
62 lines (53 loc) · 2.07 KB
/
extractFeature.py
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
60
61
62
# cSpell:disable
# ============================
# PACKAGES
# ============================
import argparse
import sys
import json
import time
import traceback
from helpers import log, extractAAC, extractDPC, extractPHC, extractAAIndex, extractPSSM
# ============================
# MAIN
# ============================
def main():
# Extracting feature from Dataset files
try:
# Checking for right argument from user
parser = argparse.ArgumentParser(
description='Extracting Features from Dataset')
parser.add_argument('feature', type=str, help='Feature to Extract')
parser.add_argument(
'dataset', type=str, help='Dataset from which the feature would be Extracted from (trainTest or Independent)')
parser.parse_args()
# Logging start of Program
feature = parser.parse_args().feature
dataset = parser.parse_args().dataset
log(time.strftime("%m/%d/%Y, %H:%M:%S", time.localtime()))
# Features
# AAC (Amino-acid Composition)
if feature == 'aac':
extractAAC(dataset, feature)
log("AAC being extracted from [ "+dataset+" ] successfully!!!")
# DPC (Di-peptid composition)
elif feature == 'dpc':
extractDPC(dataset, feature)
log("DPC being extracted from [ "+dataset+" ] successfully!!!")
# PHC (Physico-Chemical Composition)
elif feature == 'phc':
extractPHC(dataset, feature)
log("PHC being extracted from [ "+dataset+" ] successfully!!!")
# AAIndex (BioChemical Composotion)
elif feature == 'aaindex':
extractAAIndex(dataset, feature)
log("AAIndex being extracted from [ "+dataset+" ] successfully!!!")
# PSSM (Position)
elif feature == 'pssm':
extractPSSM(dataset, feature)
log("PSSM being extracted from [ "+dataset+" ] successfully!!!")
except Exception:
log("[ "+feature+" ] Feature Extraction Process Failed!!")
log(traceback.format_exc())
if __name__ == "__main__":
main()