I’m somehow stuck on the very simple task of creating the model.
I’m getting the error:
AssertionError: Wrong values for d[‘Y_prediction_test’]. [[1. 0. 0.]] != [[1. 1. 0.]]
All my previous unit test passed, no not quite sure what I’m doing wrong.
Here is the general outline of my code:

{moderator edit - solution code removed}

Could anyone suggest some ways to debug this?

Sorry, shortly after posting this I found a typo in by predict function where I used 1 as an index instead of i

Glad to hear that you found the problem. I think you should also consider calling initialize_with_zeros from model instead of directly calling np.zeros. Your version works, but then what was the point of writing that other function if you don’t call it?