GCN object has no attribute classifier?

新手上路,请多包涵
class GCN(torch.nn.Module):
    def __init__(self):
        super().__init__()
        torch.manual_seed(1234)
        self.conv1 = GCNConv(dataset.num_features, 4)
        self.conv2 = GCNConv(4, 4)
        self.conv3 = GCNConv(4, 2)

    def forward(self, x, edge_index):
        h = self.conv1(x, edge_index)
        h = h.tanh()
        h = self.conv2(h, edge_index)
        h = h.tanh()
        h = self.conv3(h, edge_index)
        h = h.tanh()

        # 分类层
        out = self.classifier(h)

        return out, h
Traceback (most recent call last):
  File "C:\Users\Administrator\Desktop\projects\test.py", line 70, in <module>
    _, h = model(data.x, data.edge_index)
  File "C:\Program Files\Python38\lib\site-packages\torch\nn\modules\module.py", line 1190, in _call_impl
    return forward_call(*input, **kwargs)
  File "C:\Users\Administrator\Desktop\projects\test.py", line 63, in forward
    out = self.classifier(h)
  File "C:\Program Files\Python38\lib\site-packages\torch\nn\modules\module.py", line 1265, in __getattr__
    raise AttributeError("'{}' object has no attribute '{}'".format(
AttributeError: 'GCN' object has no attribute 'classifier'
阅读 1.2k
撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题