CoLab에서 pytorch-geometric 환경 설정하기

import torch def format_pytorch_version(version): return version.split(‘+’)[0] TORCH_version = torch.__version__ TORCH = format_pytorch_version(TORCH_version) def format_cuda_version(version): return ‘cu’ + version.replace(‘.’, ‘’) CUDA_version = torch.version.cuda CUDA = format_cuda_version(CUDA_version) !pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-{TORCH}+{CUDA}.html !..

• format_list_bulleted 💬 LANGUAGES/Python
• · 2023. 4. 18.
• textsms

[Python/PyTorch] Sequential()을 forward()에 사용하기

문제 class GNNs(torch.nn.Module): def __init__(self, dataset): super().__init__() self.data = dataset self.conv1 = GCNConv(dataset.num_node_features, 32) self.conv2 = GCNConv(32, dataset.num_classes) def forward(self, data): x, edge_index = data.x, data.edge_index x = self.conv1(x, edge_index) x = F.leaky_relu(x) x = self.conv2(x, edge_index) return F.softmax(x, dim=1) cora_model = GNNs(cora_datas..

• format_list_bulleted 💬 LANGUAGES/Python
• · 2023. 3. 21.
• textsms

[Python/PyTorch] GCN을 사용한 모델 만들고 평가하기(accuracy)

들어가기에 앞서 이 글은 신원용교수님 수업의 과제를 수행하며 정리한 글임을 밝힙니다. 데이터 준비 1. 필요한 라이브러리 import import torch import torch.nn.functional as F from torch_geometric.datasets import Planetoid, CoraFull, Actor from torch_geometric.nn import GCNConv 2. 데이터 셋 불러오기 # Load the Cora dataset cora_dataset = Planetoid(root='/tmp/Cora', name='Cora') cora_data = cora_dataset[0] # Load the Corafull dataset corafull_..

• format_list_bulleted 💬 LANGUAGES/Python
• · 2023. 3. 20.
• textsms