蓝桥云课
|
IT学习平台
首页
课程
知识库
论坛
登录
讨论区
课程问答
关于其中GNN分类里test函数中的pred取值的问题
关于其中GNN分类里test函数中的pred取值的问题
0回复
21查看
Zheng Nan10000001
2023-05-17 06:53
全部回复
还没有人回复，沙发空缺中~
登录
后回复帖子
我要发帖
×
Close
邀请好友，双方都可获赠实验豆！
登录
后邀请好友注册，您和好友将分别获赠3个实验豆！
×
Close
注意
0% Complete
加载中，精彩就在后面...
卡住了？点击重试
×
发帖
标题
描述
编辑
预览
截取实验桌面
上传本地图片
Markdown 语法
板块
×
发帖
标题
描述
编辑
预览
截取实验桌面
上传本地图片
Markdown 语法
GNN分类的测试函数，为什么与之前几个实验写的是不一样的呢？ ``` def test(model,test_loader): model.eval() # 验证模式 correct = 0 for data in test_loader: output = model(data.x, data.edge_index, data.batch) pred = output.max(dim=1)[1] correct += pred.eq(data.y).sum().item() return correct / len(test_loader.dataset) # 准确率 ``` 这其中的 ``` pred = output.max(dim=1)[1] ``` 这一句，与之前的区别在哪里呢？为什么要这么做呢？
板块
确定删除
删除后不可恢复
×
登录
注册
下次自动登录
忘记密码？
GNN分类的测试函数，为什么与之前几个实验写的是不一样的呢？ ``` def test(model,test_loader): model.eval() # 验证模式 correct = 0 for data in test_loader: output = model(data.x, data.edge_index, data.batch) pred = output.max(dim=1)[1] correct += pred.eq(data.y).sum().item() return correct / len(test_loader.dataset) # 准确率 ``` 这其中的 ``` pred = output.max(dim=1)[1] ``` 这一句，与之前的区别在哪里呢？为什么要这么做呢？
