如何在tensorflow中读取google bucket中的文件？

import tensorflow as tf

bucket_path = 'gs://your-bucket-name'
file_pattern = 'your-file-pattern.csv'
column_names = ['col1', 'col2', 'col3']
column_defaults = [tf.float32, tf.float32, tf.float32]

dataset = tf.data.experimental.CsvDataset(
    bucket_path + '/' + file_pattern,
    record_defaults=column_defaults,
    header=True,
    select_cols=[0, 1, 2]
)

def preprocess_data(col1, col2, col3):
    # Perform necessary preprocessing
    col1 = tf.square(col1)
    col2 = tf.square(col2)
    col3 = tf.square(col3)
    return col1, col2, col3

dataset = dataset.map(preprocess_data)

for data in dataset:
    col1, col2, col3 = data
    # Process the data as needed
    print(col1, col2, col3)