It appears your code is outputting a vector of six values for that test, when the expected size is one value (for that test).
1 Like
Thanks mate
I have the same exact problem How did you fix the problem?
Maybe you are not using initialize_with_zeros
function to initialize parameters. You need to use it.
I found the issue. I should not equate variables in the optimize function to anything. i was equating number of iterations to 100 and learning rate to 0.009 etc. basically copy pasted the exercise function from before.