/
main.py
38 lines (33 loc) · 802 Bytes
/
main.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
from Experiments import *
from DataLoader import DataLoader
if __name__ == '__main__':
# load data
dl_1 = DataLoader('data\\UCI-bank-marketing.csv', 'outputs\\Marketing', 'Marketing')
dl_1.load_data()
# run classifier
ANN(dl_1)
BOOST(dl_1)
SVM_RBF(dl_1)
SVM_linear(dl_1)
KNN(dl_1)
DT(dl_1)
# load data
dl_2 = DataLoader('data\\Heart.csv', 'outputs\\Heart', 'Heart')
dl_2.load_data()
# run classifier
ANN(dl_2)
BOOST(dl_2)
SVM_linear(dl_2)
KNN(dl_2)
SVM_RBF(dl_2)
DT(dl_2)
# # load data
# dl_3 = DataLoader('data\\Cancer.csv', 'outputs\\Cancer', 'Cancer')
# dl_3.load_data()
# # run classifier
# ANN(dl_3)
# BOOST(dl_3)
# SVM_RBF(dl_3)
# SVM_linear(dl_3)
# KNN(dl_3)
# DT(dl_3)