二分类问题手动给mat数据打标签

import os
import scipy.io as sio

# 定义包含MAT文件的文件夹路径
folder_path = '你要处理的mat文件所在的文件夹路径'

# 获取文件夹中的所有MAT文件
mat_files = [f for f in os.listdir(folder_path) if f.endswith('.mat')]

# 创建一个空标签字典，用于存储每个MAT文件的标签
labels_dict = {}

# 循环处理每个MAT文件
for mat_file in mat_files:
    # 构建MAT文件的完整路径
    mat_file_path = os.path.join(folder_path, mat_file)

    # 读取MAT文件
    mat_data = sio.loadmat(mat_file_path)

    # 获取MAT文件中的数据
    data = mat_data['data']  # 替换为实际的数据变量名

    # 通过输入来手动添加标签
    label = input(f"为MAT文件 '{mat_file}' 添加标签 (1 或 0): ")

    # 将标签存储在标签字典中，以文件名作为键
    labels_dict[mat_file] = int(label)

# 选择要保存标签文件的文件夹路径
output_folder = '你要保存标签文件的目标文件夹路径'
output_path = os.path.join(output_folder, 'labels.txt')

# 打印每个MAT文件的标签
for mat_file, label in labels_dict.items():
    print(f"MAT文件: {mat_file}, 标签: {label}")

# 保存标签到指定文件夹中的文件
with open(output_path, 'w') as file:
    for mat_file, label in labels_dict.items():
        file.write(f"{mat_file}: {label}\n")

print(f"标签已保存到 '{output_path}' 文件中.")