为了使用TensorFlow和Keras训练简单的GAN ,导入Galaxy Zoo数据集的正确方法是什么?例如,是否有与导入MNIST相同的简便方法?
from keras.datasets import mnist
您必须先下载数据并对其进行预处理,然后再馈入网络。如果您转到kaggle链接并转到笔记本,则可以看到其他人如何预处理他们的数据以进行培训
重要的是您要训练什么以及要分类什么。如果要训练图像以对csv中的全部或某些类进行分类,则需要将数据作为图像,将分类作为标签。如果要使用某种分类来预测另一种分类,则某些类别将是数据,而其他类别将是您要预测的标签。
我建议将培训csv导入pandas数据框
import pandas as pd
df = pd.read_csv('../input/44352/training_solutions_rev1.csv')
然后,您将不得不为图像创建一个自定义生成器,因为它们不是在文件夹结构中按类组织的。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句