R2 = r2_score(test_labels.cpu(), pre_array)
时间: 2024-04-18 15:26:16 浏览: 70
imagenet_slim_labels.txt
R2 = r2_score(test_labels.cpu(), pre_array) 是用来计算预测结果与真实标签之间的R方值的函数。R方值用于评估回归模型的拟合程度,其取值范围在0到1之间,越接近1表示模型拟合效果越好。
其中,test_labels是真实标签的数据集,pre_array是预测结果的数据集。该函数会根据这两个数据集计算出对应的R2值,并返回结果。
阅读全文