从tfrecords中的不平衡数据生成平衡批

Question

我有一个包含20k个阳性实例和300k个阴性实例的数据集。

我想为每个类创建一个tfrecord文件，然后重新整理数据并生成具有平衡类的批处理。