PyTorch Geometric

PyTorch Geometric é uma biblioteca de extensão de aprendizagem profunda geométrica para PyTorch. O aprendizado profundo geométrico visa construir redes neurais que possam aprender com dados não euclidianos, como grafos. Abordagens de aprendizado de máquina baseado em rede gráfica têm despertado interesse crescente na modelagem de redes sociais e nos campos biomédicos, especificamente na descoberta de medicamentos. PyTorch Geometric fornece uma biblioteca fácil de usar para projetar problemas complicados de rede de grafos, como a representação da estrutura de proteínas. Possui suporte para GPU e CPU e inclui uma boa coleção de algoritmos de aprendizado de máquina baseados em grafos fundamentados em pesquisas recentes.