Unq_05 base risks

Course Q&A AI For Medicine AI For Medical Treatment
1

Hi, I am stuck on exercise 5.
I am using the method X.loc[:,‘TRTMT’]=0 to set the values of the TRTMT column and passing X to the predict_proba(X) function to get the probabilities.
However I keep getting this error:

image
image934×483 8.42 KB

What i’m doing wrong?

2

Hello @Sebastian_Miranda

Could share exercise 5 codes by personal DM. Click on my name and then message.

Regards
DP

3

Already resolved, thank you very much

1 Like
4

Kindly let learners know how you resolve your issue.

5

I was taking all the predictions instead of taking only the death probability.