def predict(self, _x_val):
mr = None
if type(_x_val) is not list:
mr = LinearRegression(self.X, self.Y)
else:
mr = MultipleRegression(self.X, self.Y)
mr.calculateCoeff()
y_val = mr.predict(_x_val)
prob_val = sigmoid(y_val)
if prob_val < 0.5:
return 0
else:
return 1
