alexeygrigorev / mlbookcamp-code Goto Github PK

Hello, not sure what the preferred way is to flag this bug. Fork vs Issue? Anyway.

1. Go to session 5
2. Scroll down to the roc_auc_score cell.
3. Add y_test = df_test.churn.values on line 3 as per below.

dv, model = train(df_full_train, df_full_train.churn.values, C=1.0)
y_pred = predict(df_test, dv, model)
y_test = df_test.churn.values

auc = roc_auc_score(y_test, y_pred)
auc

Thanks.