-
Notifications
You must be signed in to change notification settings - Fork 0
/
runner.py
82 lines (49 loc) · 1.57 KB
/
runner.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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
# -*- coding: utf-8 -*-
"""
Created on Mon Aug 19 15:52:57 2019
@author: Dipto
"""
from LogReg import *
from SVM import *
from ANN import *
if __name__ == "__main__":
n = input('Enter choice of operation: ')
if n=='EDA':
#Statistical Analysis
stats()
#Distribution of Important features of CAD
maleAndFemale()
percentOfCad()
cadPatientsByAge()
cadPatientsBySex()
cadPatientsByFH()
cadbyCS()
cadbyES()
cadbyObesity()
cadbyTCP()
cadbyVHD()
cadPatientsByHTN()
cadPatientsByBP()
cadPatientsByDM()
elif n == 'Prepare':
before_res()
after_res()
cadbyecp()
elif n == 'LR':
lr = LR(X_train,X_train_res,X_test,y_train,y_train_res,y_test)
lr.logreg_imbal()
lr.logreg_bal()
lr.roc_auc_logreg_imbal()
lr.roc_auc_logreg_bal()
elif n == 'SVM':
svm = SVM(X_train,X_train_res,X_test,y_train,y_train_res,y_test)
svm.svm_imbal()
svm.roc_auc_svm_imbal()
svm.svm_bal()
svm.roc_auc_svm_bal()
elif n == 'ANN':
ann = ANN(X_train,X_train_res,X_test,y_train,y_train_res,y_test)
ann.ann_imbal()
ann.roc_auc_ann_imbal()
ann.ann_bal()
ann.roc_auc_ann_bal()