ROC Curves and data from tf_dataset

TensorFlow General Discussion
,
1

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

2 Likes
2

@Erin , There is an entire guide here in documentation on splitting slicing data when using tensorflow_datasets.

2 Likes