First, why need to flatten X_train? Please follow the instructions and do not introduce anything new.
Secondly, num_iterations and learning_rate shouldn’t be 100 and 0.005, respectively. Check this recent post.
First, why need to flatten X_train? Please follow the instructions and do not introduce anything new.
Secondly, num_iterations and learning_rate shouldn’t be 100 and 0.005, respectively. Check this recent post.