Python 计算 R² 值并保存至文本文件

使用 Python 计算 R² 值并保存至文本文件

该代码展示如何计算每组数据的 R² 值，并将结果保存至名为 'result.txt' 的文本文件。

# 计算 R² 值
R_squared = 1 - ((c_pred[0] - c_true[0]) ** 2 + (d_pred[0] - d_true[0]) ** 2 + (e_pred[0] - e_true[0]) ** 2 + (f_pred[0] - f_true[0]) ** 2 + (g_pred[0] - g_true[0]) ** 2 + (h_pred[0] - h_true[0]) ** 2 + (i_pred[0] - i_true[0]) ** 2 + (j_pred[0] - j_true[0]) ** 2 + (k_pred[0] - k_true[0]) ** 2) ** 2 / ((c_pred[0] - 1 / c_true[0]) ** 2 + (d_pred[0] - 1 / d_true[0]) ** 2 + (e_pred[0] - 1 / e_true[0]) ** 2 + (f_pred[0] - 1 / f_true[0]) ** 2 + (i_pred[0] - 1 / i_true[0]) ** 2 + (j_pred[0] - 1 / j_true[0]) ** 2 + (k_pred[0] - 1 / k_true[0]) ** 2) ** 2

print('R² 的值为:', R_squared)

# 将结果保存至文本文件
import numpy as np

B1 = np.zeros((len(c_pred), 1))
B1[:, 0] = R_squared

np.savetxt('result.txt', B1, fmt='%.4f')

解释:

• 该代码首先使用一个公式计算 R² 值。
• 然后使用 print() 函数输出 R² 值。
• 代码使用 numpy 库创建一个矩阵 B1，并将 R² 值存储在该矩阵中。
• 最后使用 np.savetxt() 函数将矩阵 B1 保存至名为 'result.txt' 的文本文件。

注意:

• 该代码假设您已经定义了变量 c_pred, c_true, d_pred, d_true 等。
• 您需要根据您的实际情况修改代码中的变量名称和公式。
• fmt='%.4f' 用于将保存至文件的值格式化为保留小数点后四位。