Python 为CNN中的列车数据分配标签

我是kaggle的新手，我想知道如何为我的测试数据分配标签

我的train和test文件夹都具有以下格式

1
2
3
etc...

train.txt包含如下标签：

train/0.jpg 5
train/1.jpg 1
train/2.jpg 10
train/3.jpg 2
train/4.jpg 22
train/5.jpg 3
etc...

---

我知道我需要循环，并为每个培训图像分配课程编号。但不确定如何，我在网上查看过，但数据的格式不同。

首先读取.txt文件并制作路径列表和标签：

with open('train.txt', 'r') as f:
     path_label = f.readlines() # returns a list

# seperate paths and labels
paths = [i.strip().split()[0] for i in path_label] 
labels = [i.strip().split()[1] for i in path_label]

---

然后，您可以读取数据并将其提供给模型。

如果您定义了确切的问题，那么我肯定会做得更好。我现在已经循环了每一行，将路径和标签从每一行中分离出来，现在应该能够将标签分配给每一个类名。谢谢你的帮助