Circling back on this, I found that the issue was the dimension of the output… In my original post, X was of dimension d x n and y was dimension n x 1. As a side effect of changing y to one hot encoding, I corrected the dimension to 2 x n.
Circling back on this, I found that the issue was the dimension of the output… In my original post, X was of dimension d x n and y was dimension n x 1. As a side effect of changing y to one hot encoding, I corrected the dimension to 2 x n.