如何在深度学习中管理过多的课程？

Question

我正在研究人脸识别项目，并且有一个名为Multi-pie的数据集。该数据集包含317个主题。所以我的模态课有317节课我曾对数据集进行过预处理。但是我无法成功实现标签。我创建了一个包含文件名和subjectNo的csv文件。

  labels = dataset[:,1:]
  filenames = dataset[:,0:1]
  print(filenames[0])
  print(labels[0])

  filenames_shuffled, y_labels_one_hot_shuffled = shuffle(filenames, labels)

  X_train_filenames, X_val_filenames, y_train, y_val = train_test_split(
  filenames_shuffled, y_labels_one_hot_shuffled, test_size=0.2, random_state=1)

  print(X_train_filenames.shape) 
  print(y_train.shape)           

  print(X_val_filenames.shape)   
  print(y_val.shape)

此代码的输出：

  (60288, 1)
  (60288, 1)
  (15072, 1)
  (15072, 1)

但是我认为Output会这样：

  (60288, )
  (60288, 317)
  (15072, )
  (15072, 317)

我在做什么？我在下面的图片中显示了csv文件。