config.json

{
  "architectures": [
    "UnifiedCellClassifier"
  ],
  "dtype": "float32",
  "macro_to_sub": {
    "0": "B_cells_classifier",
    "1": "CD4plus_T_cells_classifier",
    "4": "Myeloid_cells_classifier",
    "5": "NK_cells_classifier",
    "7": "TRAV1_2_CD8plus_T_cells_classifier",
    "8": "gd_T_cells_classfier"
  },
  "main_classifier_config": {
    "dropout_rate": 0.2,
    "hidden_dims": [
      512,
      256
    ],
    "input_dim": 3072,
    "output_dim": 9,
    "use_residual_in_hidden": true
  },
  "main_labels": {
    "0": "B cells",
    "1": "CD4+ T cells",
    "2": "DN T cells",
    "3": "MAIT cells",
    "4": "Myeloid cells",
    "5": "NK cells",
    "6": "Progenitor cells",
    "7": "TRAV1-2- CD8+ T cells",
    "8": "gd T cells"
  },
  "model_type": "scBloodClassifier",
  "sub_classifier_names": [
    "B_cells_classifier",
    "CD4plus_T_cells_classifier",
    "Myeloid_cells_classifier",
    "NK_cells_classifier",
    "TRAV1_2_CD8plus_T_cells_classifier",
    "gd_T_cells_classfier"
  ],
  "sub_classifiers_config": {
    "B_cells_classifier": {
      "dropout_rate": 0.2,
      "hidden_dims": [
        3072,
        1536,
        768
      ],
      "input_dim": 3072,
      "output_dim": 9,
      "use_residual_in_hidden": true
    },
    "CD4plus_T_cells_classifier": {
      "dropout_rate": 0.2,
      "hidden_dims": [
        3072,
        1536,
        768
      ],
      "input_dim": 3072,
      "output_dim": 16,
      "use_residual_in_hidden": true
    },
    "Myeloid_cells_classifier": {
      "dropout_rate": 0.2,
      "hidden_dims": [
        3072,
        1536,
        768
      ],
      "input_dim": 3072,
      "output_dim": 4,
      "use_residual_in_hidden": true
    },
    "NK_cells_classifier": {
      "dropout_rate": 0.2,
      "hidden_dims": [
        3072,
        1536,
        768
      ],
      "input_dim": 3072,
      "output_dim": 6,
      "use_residual_in_hidden": true
    },
    "TRAV1_2_CD8plus_T_cells_classifier": {
      "dropout_rate": 0.2,
      "hidden_dims": [
        3072,
        1536,
        768
      ],
      "input_dim": 3072,
      "output_dim": 12,
      "use_residual_in_hidden": true
    },
    "gd_T_cells_classfier": {
      "dropout_rate": 0.2,
      "hidden_dims": [
        3072,
        1536,
        768
      ],
      "input_dim": 3072,
      "output_dim": 5,
      "use_residual_in_hidden": true
    }
  },
  "sub_labels": {
    "B_cells_classifier": {
      "0": "Activated",
      "1": "Atypical memory",
      "2": "CD5+ B cells",
      "3": "Naive",
      "4": "Naive-IFN",
      "5": "Non-switched memory",
      "6": "Plasma cells",
      "7": "Switched memory",
      "8": "Transitional"
    },
    "CD4plus_T_cells_classifier": {
      "0": "Exhausted-like memory",
      "1": "HLA-DR+ memory",
      "10": "Th2",
      "11": "Th22",
      "12": "Treg KLRB1+RORC+",
      "13": "Treg cytotoxic",
      "14": "Treg memory",
      "15": "Treg naive",
      "2": "Naive",
      "3": "Naive-IFN",
      "4": "Temra",
      "5": "Terminal effector",
      "6": "Tfh",
      "7": "Th1",
      "8": "Th1/Th17",
      "9": "Th17"
    },
    "Myeloid_cells_classifier": {
      "0": "Classical monocytes",
      "1": "Non-classical monocytes",
      "2": "cDCs",
      "3": "pDCs"
    },
    "NK_cells_classifier": {
      "0": "CD56bright",
      "1": "CD56dim CD57+",
      "2": "CD56dim CD57-",
      "3": "CD56dim CD57int",
      "4": "CD56dim CD57low",
      "5": "Proliferative"
    },
    "TRAV1_2_CD8plus_T_cells_classifier": {
      "0": "HLA-DR+",
      "1": "NKT-like",
      "10": "Tmem KLRC2+",
      "11": "Trm",
      "2": "Naive",
      "3": "Naive-IFN",
      "4": "Proliferative",
      "5": "Tcm CCR4+",
      "6": "Tcm CCR4-",
      "7": "Tem GZMB+",
      "8": "Tem GZMK+",
      "9": "Temra"
    },
    "gd_T_cells_classfier": {
      "0": "Vd1 GZMB+",
      "1": "Vd1 GZMK+",
      "2": "Vd2 GZMB+",
      "3": "Vd2 GZMK+",
      "4": "gd naive"
    }
  },
  "transformers_version": "4.56.1"
}