Можно ли преобразовать GraphSAGE для решения проблемы регрессии?
Я пытался решить проблемы классификации узлов с помощью GraphSAGE, и он хорошо работает с набором данных Cora. Теперь я хочу преобразовать его для решения проблемы регрессии, и я только меняю функцию потерь наtorch.nn.MSE()без каких-либо изменений в других частях кода.
Однако потери не уменьшаются. Итак, мне интересно, можно ли использовать GraphSAGE для решения проблемы регрессии и как мне его преобразовать?
пс:
Мой код взят с этого сайта:https://github.com/ytchx1999/GraphSAGE-Cora-Citeseer-Pubmed/tree/main
И я только меняю набор данных и конвертируюcriterion = nn.CrossEntropyLoss().to(DEVICE)кcriterion = nn.MSE().to(DEVICE).