-
Notifications
You must be signed in to change notification settings - Fork 0
/
driver.py
36 lines (29 loc) · 864 Bytes
/
driver.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
import svm
import make_granules
import read_data
import numpy as np
def main():
data = []
inp = []
num_iter = int ( input("Enter the number of iterations") )
for i in range(0,num_iter):
inp = make_granules.make_granules(i,data[:2])
data = svm.svm(inp)
print("clf is ",data[-1] )
print("***********************************************************************")
print("ON THE FINAL DATA !!!!")
df = read_data.read_data()
number_of_cols = len(df.columns)
X = df.values
Y = X[:,-1]
X = X[:,:-1]
indices = np.argwhere(Y==1)
clf = data[-1]
indices = indices.ravel()
X = X[indices]
predictions = clf.predict(X)
correctly_done = np.sum(predictions)
print("Correctly classified minority points ",correctly_done/len(indices))
print("***********************************************************************")
if __name__ == "__main__":
main()