napistu_torch.models

Neural network models and architectures for napistu-torch.

This subpackage provides model encoders, decoders, and complete architectures specifically designed for molecular network analysis and graph-based machine learning.

Modules

constants

edge_encoder

Edge encoder for Napistu-Torch.

head_utils

Utility functions supporting a subset of heads.

heads

Prediction heads for Napistu-Torch.

message_passing_encoder

Graph Neural Network models for Napistu-Torch.

node2vec

Node2Vec utilities for Napistu-Torch.