Hugging Face's logo

Bgoood
/
SpatialGT-MouseStroke-Sham

Feature Extraction
Transformers
Safetensors
PyTorch
English
spatial-transcriptomics
graph-transformer
gene-expression
finetuned
mouse-stroke
Model card Files
xet
Community
SpatialGT-MouseStroke-Sham
File size: 1,842 Bytes
7680b76
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
 
---
license: mit
tags:
  - spatial-transcriptomics
  - graph-transformer
  - gene-expression
  - finetuned
  - mouse-stroke
  - pytorch
language:
  - en
library_name: transformers
pipeline_tag: feature-extraction
---

# SpatialGT Finetuned Model - Mouse Stroke (Sham)

## Model Description

This is the **finetuned checkpoint** of SpatialGT on mouse stroke Sham (control) spatial transcriptomics data.

This model is specifically finetuned for the mouse stroke perturbation simulation case study, trained on the Sham1-1 slice.

## Model Details

- **Base Model**: SpatialGT Pretrained
- **Finetuning Data**: Mouse stroke Sham1-1 slice (Visium)
- **Finetuning Strategy**: Full finetuning (8 transformer layers unfrozen)
- **Epochs**: 100
- **Learning Rate**: 1e-4

## Usage

```python
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 mouse brain tissue
- Simulating perturbation effects in stroke-affected regions
- Comparative analysis with PT (stroke) model

## Files

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

## Related Models

- [SpatialGT-Pretrained](https://huggingface.co/Bgoood/SpatialGT-Pretrained): Base pretrained model
- [SpatialGT-MouseStroke-PT](https://huggingface.co/Bgoood/SpatialGT-MouseStroke-PT): Finetuned on PT (stroke) slice

## Citation

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

## License

MIT License