ephysatlas_model.json

{
  "model_id": "ea-encoder-channel",
  "task": "spatial-encoding",
  "model_class": "NeighborInpaintingModel",
  "vintage": "2026_W26",
  "training": {
    "random_seed": 0
  },
  "environment": {
    "python": "3.12.10",
    "xgboost": "3.4.0",
    "scikit_learn": "1.8.0",
    "numpy": "2.4.6",
    "pandas": "3.0.5",
    "torch": "2.13.0",
    "ephysatlas": "0.7.0"
  },
  "method": "transformer",
  "granularity": "channel",
  "artifacts": {
    "weights": "spatial_encoder.pt",
    "context": [
      "agea_vol_pca.npy",
      "merfish_vol_pca.npy"
    ],
    "confidence": "confidence_model.pt",
    "neighbor_bank": "neighbor_bank.npz"
  },
  "inputs": {
    "table": "channels.pqt",
    "index": [
      "pid",
      "channel"
    ],
    "columns": [
      "x",
      "y",
      "z"
    ]
  },
  "outputs": {
    "kind": "continuous",
    "columns": [
      "rms_lf",
      "psd_lfp",
      "psd_alpha",
      "psd_beta",
      "psd_gamma",
      "psd_delta",
      "psd_theta",
      "psd_lfp_csd_diff1",
      "psd_alpha_csd_diff1",
      "psd_beta_csd_diff1",
      "psd_gamma_csd_diff1",
      "psd_delta_csd_diff1",
      "psd_theta_csd_diff1",
      "rms_lf_csd_diff1",
      "psd_residual_lfp",
      "psd_residual_alpha",
      "psd_residual_beta",
      "psd_residual_gamma",
      "psd_residual_delta",
      "psd_residual_theta",
      "decay_fit_error",
      "decay_fit_r_squared",
      "decay_n_peaks",
      "aperiodic_exponent",
      "aperiodic_offset",
      "cor_ratio",
      "rms_ap",
      "alpha_mean",
      "alpha_std",
      "spike_count",
      "tip_time_secs",
      "recovery_time_secs",
      "peak_time_secs",
      "trough_time_secs",
      "trough_val",
      "tip_val",
      "peak_val",
      "recovery_slope",
      "depolarisation_slope",
      "repolarisation_slope",
      "polarity"
    ],
    "feature_order_sha256": "ebc1285d9121753c4adf0f96ff456df3c0afd1ac4124ffa3556907283489efaa"
  },
  "config": {
    "architecture": {
      "f_ctx": 100,
      "f_ephys": 41,
      "f_out": 41,
      "d_model": 128,
      "nhead": 8,
      "depth": 2,
      "drop": 0.15
    },
    "neighbourhood": {
      "radius_um": 500.0,
      "m_max": 8,
      "selection": "nearest",
      "allow_same_probe": false
    },
    "context": {
      "n_cell_pcs": 50,
      "n_gene_pcs": 50,
      "voxel_um": 200
    }
  }
}