Upload config.json with huggingface_hub

config.json

@@ -0,0 +1,258 @@
+{
+  "version": "1.0.0",
+  "description": "Stellar Foundation Models - Parameter Estimation and Cross-Modal Retrieval",
+  "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 Å)",
+        "architecture": "SpecFormer"
+      }
+    },
+    "specclip_models": {
+      "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"
+      }
+    },
+    "parameter_estimation": {
+      "lamost_lrs": {
+        "sbi": {
+          "age": {
+            "path": "downstream_models/lrs/sbi/age.pkl",
+            "parameters": ["age"],
+            "method": "SBI",
+            "stats": null
+          },
+          "nu_max": {
+            "path": "downstream_models/lrs/sbi/numax.pkl",
+            "parameters": ["nu_max"],
+            "method": "SBI",
+            "stats": null
+          },
+          "mass": {
+            "path": "downstream_models/lrs/sbi/mass.pkl",
+            "parameters": ["mass"],
+            "method": "SBI",
+            "stats": null
+          },
+          "rad": {
+            "path": "downstream_models/lrs/sbi/rad.pkl",
+            "parameters": ["rad"],
+            "method": "SBI",
+            "stats": null
+          },
+          "dnu": {
+            "path": "downstream_models/lrs/sbi/dnu.pkl",
+            "parameters": ["dnu"],
+            "method": "SBI",
+            "stats": null
+          },
+          "rv": {
+            "path": "downstream_models/lrs/sbi/rv.pkl",
+            "parameters": ["rv"],
+            "method": "SBI",
+            "stats": null
+          },
+          "dpi1": {
+            "path": "downstream_models/lrs/sbi/DP.pkl",
+            "parameters": ["dpi"],
+            "method": "SBI",
+            "stats": null
+          },
+          "teff": {
+            "path": "downstream_models/lrs/sbi/teff.pkl",
+            "parameters": ["teff"],
+            "method": "SBI",
+            "stats": null
+          },
+          "logg": {
+            "path": "downstream_models/lrs/sbi/logg.pkl",
+            "parameters": ["logg"],
+            "method": "SBI",
+            "stats": null
+          }
+        },
+        "mlp": {
+          "feh": {
+            "path": "downstream_models/lrs/mlp/feh.pt",
+            "parameters": ["fe_h"],
+            "method": "MLP",
+            "stats": "stats/lrs/feh_mean_std.json"
+          },
+          "ebprp": {
+            "path": "downstream_models/lrs/mlp/ebprp.pt",
+            "parameters": ["e_bp_rp"],
+            "method": "MLP",
+            "stats": "stats/lrs/ebprp_mean_std.json"
+          },
+          "afe": {
+            "path": "downstream_models/lrs/mlp/afe.pt",
+            "parameters": ["a_fe"],
+            "method": "MLP",
+            "stats": "stats/lrs/afe_mean_std.json"
+          },
+          "cfe": {
+            "path": "downstream_models/lrs/mlp/cfe.pt",
+            "parameters": ["c_fe"],
+            "method": "MLP",
+            "stats": "stats/lrs/cfe_mean_std.json"
+          },
+          "nfe": {
+            "path": "downstream_models/lrs/mlp/nfe.pt",
+            "parameters": ["n_fe"],
+            "method": "MLP",
+            "stats": "stats/lrs/nfe_mean_std.json"
+          },
+          "alfe": {
+            "path": "downstream_models/lrs/mlp/alfe.pt",
+            "parameters": ["al_fe"],
+            "method": "MLP",
+            "stats": "stats/lrs/alfe_mean_std.json"
+          },
+          "mgfe": {
+            "path": "downstream_models/lrs/mlp/mgfe.pt",
+            "parameters": ["mg_fe"],
+            "method": "MLP",
+            "stats": "stats/lrs/mgfe_mean_std.json"
+          },
+          "mnfe": {
+            "path": "downstream_models/lrs/mlp/mnfe.pt",
+            "parameters": ["mn_fe"],
+            "method": "MLP",
+            "stats": "stats/lrs/mnfe_mean_std.json"
+          },
+          "nife": {
+            "path": "downstream_models/lrs/mlp/nife.pt",
+            "parameters": ["ni_fe"],
+            "method": "MLP",
+            "stats": "stats/lrs/nife_mean_std.json"
+          },
+          "ofe": {
+            "path": "downstream_models/lrs/mlp/ofe.pt",
+            "parameters": ["o_fe"],
+            "method": "MLP",
+            "stats": "stats/lrs/ofe_mean_std.json"
+          },
+          "sife": {
+            "path": "downstream_models/lrs/mlp/sife.pt",
+            "parameters": ["si_fe"],
+            "method": "MLP",
+            "stats": "stats/lrs/sife_mean_std.json"
+          },
+          "tife": {
+            "path": "downstream_models/lrs/mlp/tife.pt",
+            "parameters": ["ti_fe"],
+            "method": "MLP",
+            "stats": "stats/lrs/tife_mean_std.json"
+          },
+          "crfe": {
+            "path": "downstream_models/lrs/mlp/crfe.pt",
+            "parameters": ["cr_fe"],
+            "method": "MLP",
+            "stats": "stats/lrs/crfe_mean_std.json"
+          },
+          "cafe": {
+            "path": "downstream_models/lrs/mlp/cafe.pt",
+            "parameters": ["ca_fe"],
+            "method": "MLP",
+            "stats": "stats/lrs/cafe_mean_std.json"
+          }
+        }
+      },
+      "gaia_xp": {
+        "mlp": {
+          "afe": {
+            "path": "downstream_models/xp/mlp/afe.pt",
+            "parameters": ["a_fe"],
+            "method": "MLP",
+            "stats": "stats/xp/afe_mean_std.json"
+          },
+          "cfe": {
+            "path": "downstream_models/xp/mlp/cfe.pt",
+            "parameters": ["c_fe"],
+            "method": "MLP",
+            "stats": "stats/xp/cfe_mean_std.json"
+          },
+          "nfe": {
+            "path": "downstream_models/xp/mlp/nfe.pt",
+            "parameters": ["n_fe"],
+            "method": "MLP",
+            "stats": "stats/xp/nfe_mean_std.json"
+          },
+          "feh": {
+            "path": "downstream_models/xp/mlp/feh.pt",
+            "parameters": ["fe_h"],
+            "method": "MLP",
+            "stats": "stats/xp/FeH_mean_std.json"
+          },
+          "ebprp": {
+            "path": "downstream_models/xp/mlp/ebprp.pt",
+            "parameters": ["e_bp_rp"],
+            "method": "MLP",
+            "stats": "stats/xp/ebprp_mean_std.json"
+          }
+        }
+      },
+      "sbi": {
+        "teff": {
+          "path": "downstream_models/xp/sbi/teff.pkl",
+          "parameters": ["teff"],
+          "method": "SBI",
+          "stats": null
+        },
+        "logg": {
+          "path": "downstream_models/xp/sbi/logg.pkl",
+          "parameters": ["logg"],
+          "method": "SBI",
+          "stats": null
+        }
+      }
+    },
+    "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": "~500 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 Å",
+    "gaia_xp": "336-1020 nm"
+  },
+  "preprocessing": {
+    "lamost_lrs": "GASPP continuum normalization + log-wavelength interpolation",
+    "gaia_xp": "Linear interpolation + normalize at 550nm"
+  }
+}