def get_train_dataset():
    vecs = np.array([[1, 1], [1, 0], [0, 1], [0, 0]])
    labels = np.array([1, 0, 0, 0])
    return (vecs, labels)
def get_train_dataset():
    vecs = np.array([[1, 1], [1, 0], [0, 1], [0, 0]])
    labels = np.array([1, 0, 0, 0])
    return (vecs, labels)