如何将包含矢量和标签的Pandas转换为TensorFlow中的DataFrame输入

import pandas as pd
import tensorflow as tf

data = pd.read_csv('data.csv')  # 假设数据集保存在data.csv文件中

features = data.drop('label', axis=1)  # 假设标签列名为'label'
labels = data['label']

dataset = tf.data.Dataset.from_tensor_slices((features.values, labels.values))

batch_size = 32
dataset = dataset.batch(batch_size)
dataset = dataset.shuffle(buffer_size=len(data))
dataset = dataset.repeat()

model.fit(dataset, epochs=10, steps_per_epoch=len(data)//batch_size)