**prob_xgb = xgb_model.predict_proba(X_test) prob_xgb = prob_xgb[:, 1] auc_NB= roc_auc_score(y_test, prob_xgb) print("Accuracy on test set : {:.3f}".form