Hi,
I think there is a small technical issue in the sixth page of the first mission of neural network course. for the train function, it’s been asked to return a 2D numpy array containing weights for the linear regression model. However, based on the sklearn documentation, returns of SGDRegressor.coef_ method is a (n_features, ). I guess it’s a 1D numpy array.
Screen Link: https://app.dataquest.io/m/241/representing-neural-networks/6/fitting-a-linear-regression-neural-network