| { |
| "model_type": "unified_cell_classifier", |
| "auto_map": { |
| "AutoConfig": "configuration_unified_cell.UnifiedCellClassifierConfig", |
| "AutoModel": "unified_cell_classifier.UnifiedCellClassifier" |
| }, |
| "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" |
| ], |
| "main_classifier_config": { |
| "input_dim": 3072, |
| "hidden_dims": [512, 256], |
| "output_dim": 9, |
| "dropout_rate": 0.2, |
| "use_residual_in_hidden": true |
| }, |
| "sub_classifiers_config": { |
| "B_cells_classifier": { |
| "input_dim": 3072, |
| "hidden_dims": [3072, 1536, 768], |
| "output_dim": 9, |
| "dropout_rate": 0.2, |
| "use_residual_in_hidden": true |
| }, |
| "CD4plus_T_cells_classifier": { |
| "input_dim": 3072, |
| "hidden_dims": [3072, 1536, 768], |
| "output_dim": 16, |
| "dropout_rate": 0.2, |
| "use_residual_in_hidden": true |
| }, |
| "Myeloid_cells_classifier": { |
| "input_dim": 3072, |
| "hidden_dims": [3072, 1536, 768], |
| "output_dim": 4, |
| "dropout_rate": 0.2, |
| "use_residual_in_hidden": true |
| }, |
| "NK_cells_classifier": { |
| "input_dim": 3072, |
| "hidden_dims": [3072, 1536, 768], |
| "output_dim": 6, |
| "dropout_rate": 0.2, |
| "use_residual_in_hidden": true |
| }, |
| "TRAV1_2_CD8plus_T_cells_classifier": { |
| "input_dim": 3072, |
| "hidden_dims": [3072, 1536, 768], |
| "output_dim": 12, |
| "dropout_rate": 0.2, |
| "use_residual_in_hidden": true |
| }, |
| "gd_T_cells_classfier": { |
| "input_dim": 3072, |
| "hidden_dims": [3072, 1536, 768], |
| "output_dim": 5, |
| "dropout_rate": 0.2, |
| "use_residual_in_hidden": true |
| } |
| } |
| } |
