mxnet - MXNet的CSVIter中data_shape是什么意思

Question

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

Answer 1

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

您可以查看CSVIter 文档了解更多详细信息。