What is the best way to extract the y_test and X_test data once we have it converted into the tf_dataset?
I could get the y_test with y_tfdf = np.concatenate([y for x, y in test_ds], axis=0)
I was hoping to compare with other GBMs using plot_roc_curve() from sklearn