File size: 2,717 Bytes
b99de66 584995d bc8b7ef b99de66 584995d b99de66 02fb5f2 b99de66 02fb5f2 b99de66 584995d b99de66 584995d b99de66 584995d b99de66 bc8b7ef |
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 |
{
"version": "1.0.0-no-param",
"description": "SpecCLIP: Aligning and Translating Spectroscopic Measurements for Stars",
"founation_models": {
"encoders": {
"xp_encoder": {
"path": "encoders/xp_encoder.ckpt",
"description": "Gaia XP spectrum encoder (epoch=191)",
"input": "Gaia XP spectra (336-1020 nm)",
"architecture": "SpecFormer"
},
"xp_encoder_ae_768": {
"path": "encoders/xp_encoder_mlp.ckpt",
"description": "Gaia XP spectrum encoder with autoencoders (epoch=191)",
"input": "Gaia XP spectra (336-1020 nm)",
"architecture": "MLP"
},
"lrs_encoder": {
"path": "encoders/lrs_encoder.ckpt",
"description": "LAMOST LRS spectrum encoder (epoch=128)",
"input": "LAMOST LRS spectra (4000-5600 \u00c5)",
"architecture": "SpecFormer"
}
},
"specclip_models": {
"specclip_model_base": {
"path": "specclip/specclip_mlp.ckpt",
"description": "SpecClip cross-modal model with unified embeddings, no decoders, xp encoded with mlp (epoch=110)",
"architecture": "SpecClipModel_mlp"
},
"specclip_model_predrecon_mlp": {
"path": "specclip/specclip_model_predrecon_mlp.ckpt",
"description": "SpecClip cross-modal model with unified embeddings, both decoders, xp encoded with mlp (epoch=110)",
"architecture": "SpecClipModel_reconstruct_embed768_mlp"
},
"specclip_model_split_mlp": {
"path": "specclip/specclip_model_split_mlp.ckpt",
"description": "SpecClip cross-modal model with split embeddings, both decoders, xp encoded with mlp (epoch=110)",
"architecture": "SpecClipModel_reconstruct_split_5122562_mlp_recordloss"
}
},
"retrieval": {
"model_path": "specclip/specclip_model_split_mlp.ckpt",
"description": "Used for cross-modal retrieval and prediction"
}
},
"test_data": {
"gaia_lamost_split": {
"path": "test_data/gaia_lamost_test_only.h5",
"description": "Test set with paired Gaia XP and LAMOST LRS spectra",
"size": "~10 MB",
"contents": {
"test/source_ids": "Gaia source identifiers",
"test/gaia_spectra": "Gaia XP spectra (N, 343)",
"test/lamost_spectra": "LAMOST LRS spectra (N, 1462)"
},
"required_for": [
"retrieval",
"embedding_database"
],
"optional": true
}
},
"wavelength_ranges": {
"lamost_lrs": "4000-5600 \u00c5",
"gaia_xp": "336-1020 nm"
},
"preprocessing": {
"lamost_lrs": "GASPP continuum normalization + log-wavelength interpolation",
"gaia_xp": "Linear interpolation + normalize at 550nm"
}
}
|