def predict(self, X):
    y_pred = np.round(self.sigmoid(X.dot(self.param))).astype(int)
    return y_pred
def predict(self, X):
    y_pred = np.round(self.sigmoid(X.dot(self.param))).astype(int)
    return y_pred