I am trying to complete the predict function, but encountering with assertion error:
I used below loop in the function, to insert the predicted value into the vector:
if f_wb >= 0.5:
p[i] = 1
elif f_wb < 0.5:
p[i] = 0
Below is the error, I am getting, when I ran the test:
Output of predict: shape (4,), value [0. 0. 0. 1.]
AssertionError Traceback (most recent call last)
in
9
10 # UNIT TESTS
—> 11 predict_test(predict)
~/work/public_tests.py in predict_test(target)
69 raise ValueError(“Did you apply the sigmoid before applying the threshold?”)
70 assert result.shape == (len(X),), f"Wrong length. Expected : {(len(X),)} got: {result.shape}"
—> 71 assert np.allclose(result, expected_1), f"Wrong output: Expected : {expected_1} got: {result}"
72
73 b = -1.7
AssertionError: Wrong output: Expected : [1.0, 1.0, 1.0, 0.0, 1.0, 0.0, 0.0, 1.0] got: [0. 0. 0. 0. 0. 0. 0. 1.]
Expected output
Output of predict: shape (4,),value [0. 1. 1. 1.]
I tried a couple of ways to fix this, but no luck. Any help is greatly appreciated.