num_train_data = X_train.shape[0]
num_test_data = X_test.shape[0]
train_data = tf.data.Dataset.from_tensor_slices((X_train,y_train))
test_dataset = tf.data.Dataset.from_tensor_slices((X_test,y_test))
valid_data = test_dataset.take(int(0.5*num_test_data))
test_data = test_dataset.skip(int(0.5*num_test_data))

        __
        __