更新 function.py
Gitea Actions Experiment Test / Experiment-Actions (push) Successful in 22s

This commit is contained in:
aaa
2024-10-17 17:12:10 +08:00
parent 586eae6138
commit 558f4e3963
+1 -1
View File
@@ -74,7 +74,7 @@ def load_dataset(directory):
labels = []
for i, file_name in enumerate(file_list):
# 文件名格式:label_index.txt,例如 0_1.txt,标签为 0
# 文件名格式 label_index.txt,例如 0_1.txt,标签为 0
label = int(file_name.split('_')[0])
labels.append(label)
data_matrix[i, :] = img_to_vector(os.path.join(directory, file_name))