MXNet的CSVIter中data_shape的含义是什么？

Question

如果我有一个包含20个功能的400行输入矩阵，并且我有10个标签，那么我的data_shape和label_shape元组应该是什么

Answer 1

输入CSV文件的每一行都可以被视为一个向量，它将被重新整形为data_shape。因此，如果输入文件中的行是1,2,3,4,5,6且data_shape是（3,2），那么该行将被重新整形，从而产生数组[[1,2]，[3,4]形状（3,2）的，[5,6]]。

您可以查看CSVIter doc了解详情。