Hi. I get “ValueError: cannot reshape array of size 16 into shape (4,1)” and i cant find the mistake. Thanks in advance.

*{moderator edit - solution code removed}*

the error i get:

```
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
<ipython-input-20-7f17a31b22cb> in <module>
----> 1 model_test(model)
~/work/release/W2A2/public_tests.py in model_test(target)
109 y_test = np.array([1, 0, 1])
110
--> 111 d = target(X, Y, x_test, y_test, num_iterations=50, learning_rate=1e-4)
112
113 assert type(d['costs']) == list, f"Wrong type for d['costs']. {type(d['costs'])} != list"
<ipython-input-19-3fd161463d46> in model(X_train, Y_train, X_test, Y_test, num_iterations, learning_rate, print_cost)
7 w=params['w']
8 b=params['b']
----> 9 Y_prediction_test = predict(w, b, X_test)
10 Y_prediction_train = predict(w, b, X_train)
11 if print_cost:
<ipython-input-17-e2cb49b1cb55> in predict(w, b, X)
16 m = X.shape[1]
17 Y_prediction = np.zeros((1, m))
---> 18 w = w.reshape(X.shape[0], 1)
19
20 # Compute vector "A" predicting the probabilities of a cat being present in the picture
ValueError: cannot reshape array of size 16 into shape (4,1)
```

and the function predict that i am using:

*{moderator edit - solution code removed}*

And the full notebook: