from numpy import *
from LogisticRegression import LogisticRegression as LR
def load_train_data():
x = []
y = []
with open('./data.csv') as f:
for line in f.readlines():
fields = line.strip().split(',')
x.append([1.0, float(fields[0]), float(fields[1])])
y.append(float(fields[2]))
return mat(x), mat(y).transpose()
if __name__ == '__main__':
x, y = load_train_data()
tx, ty = x, y
lr = LR()
lr.train(x, y, {'alpha': 1, 'max_iteration': 25, 'optimize_type': 'smooth_stock_grad_descent'})
accuracy = lr.test(tx, ty)
print('accuracy is ', accuracy)
lr.inspect()
