models/replogle_k562/cell/config.json

{
  "dataset": "replogle_k562",
  "embedding": "pca",
  "split": "cell",
  "rep_mode": "gene_op",
  "match": "batch",
  "d_pert": 64,
  "hidden": 512,
  "depth": 4,
  "dropout": 0.0,
  "lr": 0.001,
  "weight_decay": 1e-05,
  "batch_size": 1024,
  "epochs": 60,
  "lam_tan": 1.0,
  "lam_semi": 0.5,
  "lam_reg": 0.0001,
  "grad_clip": 5.0,
  "train_frac": 1.0,
  "lam_dist": 0.0,
  "n_dist_perts": 4,
  "dist_n": 64,
  "seed": 0,
  "device_index": 3,
  "components": [
    "map",
    "tan",
    "semi"
  ]
}