import numpy as np

# FEMALE 数据
female_data = {
    "Petite_Slim": np.array([-2.405284, -1.5530001, -0.24272324, -1.443513, -0.90185016, 1.2732644,
                             -0.36933944, -0.05060475, -2.3866267, -0.5039462]),
    "Average_Fit": np.array([-1.3060766, -1.6338427, 0.70017195, -0.17132899, -1.1940454, 0.29538375,
                             0.14818205, 0.6067093, -1.2169323, -0.39638677]),
    "Athletic_Tall": np.array([0.22818007, -1.1036383, -1.5221908, 0.73692316, -0.35802472, 0.948243,
                               -0.03018424, 0.55922085, -0.16469094, -1.558347]),
    "Curvy_Plus": np.array([-0.5351595, -0.6430263, 1.4178932, 0.7848561, 0.29656753, 0.00878574,
                            0.9825444, -0.5219289, 0.40674126, 0.18135245]),
    "Big_and_Heavy": np.array([1.8121121, 0.3021033, 1.3633814, 1.6494894, 2.1491473, 0.9186887,
                               0.9121304, -1.3406857, 2.321915, -1.0636625])
}

# MALE 数据
male_data = {
    "Male_Skinny_Short": np.array([-0.7108757, -0.7559417, -0.94977564, -0.5261638, -1.7385905, -0.6821138,
                                   -3.8808734, -1.083193, -0.30184665, 0.64134985]),
    "Male_Standard": np.array([0.5232158, -0.6872696, -1.1425025, 0.7785673, -1.8967177, -0.9401324,
                               0.06393365, -0.3843205, -0.50072473, -0.5430138]),
    "Male_V_Shape": np.array([1.1519345, 0.38004422, -1.7919775, -0.37787473, -0.1616468, 0.6167449,
                              0.59436065, -0.7820431, -1.8210857, -0.37498614]),
    "Male_Stocky_Rect": np.array([0.2341475, -0.22931805, 1.9749473, -0.05317437, -1.4906491, -2.5173957,
                                  0.04375501, 0.03236455, -0.23340486, 0.4409504]),
    "Male_Giant_Heavy": np.array([2.932486, 0.4136964, 1.1478211, 1.5842791, 1.3554143, -1.7326043,
                                  1.195378, -1.1438893, 0.8729984, -0.06384684])
}

# 保存 FEMALE 数据
for i, (body_type, data) in enumerate(female_data.items()):
    filename = f"betas/FEMALE_BETA_{i}.npz"
    np.savez(filename, beta=data)
    print(f"Saved {filename} - {body_type}")

# 保存 MALE 数据
for i, (body_type, data) in enumerate(male_data.items()):
    filename = f"betas/MALE_BETA_{i}.npz"
    np.savez(filename, beta=data)
    print(f"Saved {filename} - {body_type}")

print("\n所有文件已保存完成！")

# 验证文件保存是否正确
print("\n验证保存的文件:")
for i in range(5):
    female_file = f"betas/FEMALE_BETA_{i}.npz"
    male_file = f"betas/MALE_BETA_{i}.npz"
    
    female_loaded = np.load(female_file)
    male_loaded = np.load(male_file)
    
    print(f"{female_file}: shape={female_loaded['beta'].shape}, dtype={female_loaded['beta'].dtype}")
    print(f"{male_file}: shape={male_loaded['beta'].shape}, dtype={male_loaded['beta'].dtype}")