import torch
from pretrain.model_spatialpt import SpatialNeighborTransformer
from pretrain.Config import Config

# Load configuration
config = Config()

# Initialize model
model = SpatialNeighborTransformer(config)

# Load finetuned weights
from safetensors.torch import load_file
state_dict = load_file("model.safetensors")
model.load_state_dict(state_dict)

model.eval()

Intended Use

This model is intended for:

Reconstructing gene expression in stroke-affected mouse brain tissue
Simulating perturbation effects in ischemic regions (ICA, PIA)
Comparative analysis with Sham (control) model

Files

model.safetensors: Model weights in safetensors format
training_args.bin: Training arguments

Related Models

SpatialGT-Pretrained: Base pretrained model
SpatialGT-MouseStroke-Sham: Finetuned on Sham (control) slice

Citation

If you use this model, please cite our paper (details to be added upon publication).

License

MIT License

Bgoood/SpatialGT-MouseStroke-PT

SpatialGT Finetuned Model - Mouse Stroke (PT)

Model Description

Model Details

Usage

Intended Use

Files

Related Models

Citation

License

Bgoood/
SpatialGT-MouseStroke-PT