GraPHFormer / graphformer /__init__.py
uzshah's picture
Initial commit: GraPHFormer codebase
cf84204
"""
GraPHFormer: Graph-Persistence Hybrid Transformer for Neuron Morphology
A CLIP-style contrastive learning framework for neuron representation learning
that combines tree-structured morphology with persistence images.
"""
from .models import CLIPModel, FineTuneModel, CLIPLoss
from .losses import InfoNCELoss, NTXentLoss, TripletLoss
__version__ = "1.0.0"
__all__ = ["CLIPModel", "FineTuneModel", "CLIPLoss", "InfoNCELoss", "NTXentLoss", "TripletLoss"]