Profile Encoder

This repository contains the implementation of Profile Encoder for Genotype Induced Drug Design.

Structure

eb_jepa/: EB-JEPA related code.
ijepa_model.py: I-JEPA model implementation.
vit_vae.py: ViT-VAE model implementation.
interpret_vit_vae.py: Interpretation scripts.
Training scripts: ijepa_train_script.py, mse_vit_vae_train_script.py, vit_vae_train_script.py.

Name		Name	Last commit message	Last commit date
Latest commit History 19 Commits
eb_jepa		eb_jepa
.gitignore		.gitignore
README.md		README.md
__init__.py		__init__.py
check_patient_alignment_prognosis.py		check_patient_alignment_prognosis.py
check_prognosis.py		check_prognosis.py
class_supports.txt		class_supports.txt
environment.yaml		environment.yaml
genomic_umap_3d.html		genomic_umap_3d.html
ijepa_model.py		ijepa_model.py
ijepa_train_script.py		ijepa_train_script.py
inspect_data.py		inspect_data.py
interpret_vit_vae.py		interpret_vit_vae.py
launch_sweep.sh		launch_sweep.sh
mse_vit_vae_train_script.py		mse_vit_vae_train_script.py
patient_alignment_prognosis.json		patient_alignment_prognosis.json
umap_classification_3d.py		umap_classification_3d.py
vit_vae.py		vit_vae.py
vit_vae_train_script.py		vit_vae_train_script.py
vital_status_umap3d.html		vital_status_umap3d.html