Описание
Этот репозиторий GitHub, chennnM/GCNII, предлагает реализацию на PyTorch исследовательской статьи "Simple and Deep Graph Convolutional Networks" (GCNII). Проект направлен на предоставление исследователям и разработчикам легкодоступной и функциональной кодовой базы для экспериментов с GCNII и ее развития.
Читать далее
Реализация поддерживает различные графовые наборы данных, включая эталонные наборы, такие как Cora, Citeseer и Pubmed, а также дополнительные наборы данных, такие как Chameleon, Cornell, Texas и Wisconsin. Она позволяет воспроизводить как полу-контролируемые, так и полностью контролируемые сценарии обучения, отражая экспериментальные настройки, описанные в связанных исследованиях, таких как GCN и Geom-GCN. Репозиторий также включает скрипты для выполнения этих экспериментов, что упрощает воспроизведение полученных результатов.
Ключевые компоненты реализации включают основную архитектуру модели, утилиты для загрузки и предварительной обработки данных, а также скрипты обучения. Проект подробно описывает свои зависимости, такие как CUDA, Python, PyTorch, NetworkX и scikit-learn, гарантируя, что пользователи смогут правильно настроить свою среду. Кроме того, в папке PyG (PyTorch Geometric) предоставляется эталонная реализация для тех, кто предпочитает или требует интеграции с библиотекой PyTorch Geometric, с указанием конкретных требований к версиям torch-geometric и ogb.
Репозиторий также представляет сводку результатов точности тестирования на различных наборах данных и в различных настройках, предлагая пользователям эталон для сравнения своих собственных реализаций или выводов. Инструкции по использованию четкие, направляя пользователей по выполнению предоставленных shell-скриптов для полу-контролируемых, полностью контролируемых экспериментов и индуктивных экспериментов PPI. Это делает модель GCNII доступной для практического применения в задачах графового машинного обучения.
Проект активно поддерживается, с недавними обновлениями файла README. Он подчеркивает совместный характер разработки с открытым исходным кодом, а вклады и форки указывают на вовлеченность сообщества. Основным используемым языком является Python, с небольшим процентом скриптов Shell для вспомогательных целей. Этот ресурс ценен для всех, кто интересуется передовыми графовыми нейронными сетями и их приложениями.
- КатегорияПлатформы машинного обучения
- Модель ценообразованияОткрытый код
- РазвёртываниеОблако, На своём сервере
- ПлатформыLinux, Windows
Основные функции GCNII
Реализация модели GCNII на PyTorch
Поддержка полу-контролируемого обучения
Поддержка полностью контролируемого обучения
Включает эталонные наборы данных (Cora, Citeseer, Pubmed)
Включает дополнительные наборы данных (Chameleon, Cornell, Texas, Wisconsin)
Предоставляет скрипты для воспроизведения результатов экспериментов
Предлагает эталонную реализацию PyTorch Geometric
Детализирует зависимости для настройки среды
Суммирует результаты точности тестирования
Включает инструкции по использованию для экспериментов
Кодовая база написана в основном на Python
Начало работы с GCNII
Настройка среды: Установите CUDA, Python 3.6.9, PyTorch 1.3.1, NetworkX 2.1, scikit-learn.
Загрузка наборов данных: Поместите данные в папки 'data' и 'newdata'.
Воспроизведение полу-контролируемых результатов: Запустите `sh semi.sh`.
Воспроизведение полностью контролируемых результатов: Запустите `sh full.sh`.
Воспроизведение индуктивных результатов PPI: Запустите `sh ppi.sh`.
Интеграция реализации PyG: Убедитесь, что установлены torch-geometric >= 1.5.0 и ogb >= 1.2.0.
Запуск примеров PyG: Выполните `python cora.py` или `python arxiv.py`.
Варианты использования GCNII
- Исследования графовых нейронных сетей
- Воспроизведение результатов исследований
- Графовое машинное обучение
- Бенчмаркинг моделей
- Образовательный инструмент
- Индуктивное обучение на графах








