Predict function for [logistic regression as a neural network](https://whiteviolin.medium.com/implementing-logistic-regression-as-a-neural-network-from-scratch-eff9d9cc98bc) article

predict.py

def predict(self, x):
    """
    Predicts the y values based on the training data.
    """
    prediction = []
    for single_data in x:
      prediction.append(
          1 if self.sigmoid(np.dot(single_data, self.W) + self.b) > 0.5 else 0
      )

    return prediction