如何从(批处理的) tf.data.Dataset对象确定分类频率

import tensorflow as tf
from collections import Counter

# 创建 tf.data.Dataset 对象，包含你的数据集
dataset = ...

# 数据预处理和转换
dataset = dataset.map(preprocess_fn)

# 批处理
batch_size = 32
dataset = dataset.batch(batch_size)

# 提取标签
labels_dataset = dataset.map(lambda x, y: y)

# 统计标签频率
label_counts = Counter()
for labels in labels_dataset:
    label_counts.update(labels.numpy())

# 打印每个类别的频率
for label, count in label_counts.items():
    print(f"类别 {label} 的频率为 {count}")

# 推荐的腾讯云相关产品和产品介绍链接地址
# 腾讯云产品：腾讯云机器学习平台（https://cloud.tencent.com/product/tfml）
# 腾讯云产品：腾讯云数据集成服务（https://cloud.tencent.com/product/dts）
# 腾讯云产品：腾讯云数据仓库（https://cloud.tencent.com/product/dws）