代码拉取完成,页面将自动刷新
同步操作将从 Fu_Yanbo/双色球预测 强制同步,此操作会覆盖自 Fork 仓库以来所做的任何修改,且无法恢复!!!
确定后同步将在后台操作,完成时将刷新页面,请耐心等待。
import torch
from tools import GetConfusionMatrix
def test(model, ssq_loader):
num1 = 0
num2 = 0
for data_lists, target_lists in ssq_loader:
model.eval()
pred_lists = model(data_lists)
pred_target = torch.max(pred_lists, 1)[1]
true_target = torch.max(target_lists, 1)[1]
tmp_num1 = GetConfusionMatrix(pred_target, true_target)
tmp_num2 = pred_target.numel()
num1 += tmp_num1
num2 += tmp_num2
print(num1 / num2)
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。