architectraghu commited on
Commit
ea5a92f
·
verified ·
1 Parent(s): 6666bcc

Upload training_config.json

Browse files
Files changed (1) hide show
  1. training_config.json +70 -0
training_config.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "CBC Healthcare Readmission Classifier",
3
+ "hf_repo": "careerbytecode/mlops-ref-healthcare-readmission",
4
+ "task": "binary classification (30-day hospital readmission)",
5
+ "model_type": "sigmoid-calibrated rf (sklearn Pipeline, 3-branch ColumnTransformer)",
6
+ "framework": "scikit-learn",
7
+ "serialization": "joblib",
8
+ "loader": "joblib.load -> CalibratedClassifierCV; call .predict_proba(DataFrame)[:, 1]",
9
+ "random_state": 42,
10
+ "split": {
11
+ "train": 61059,
12
+ "val": 20353,
13
+ "test": 20354
14
+ },
15
+ "feature_columns": [
16
+ "time_in_hospital",
17
+ "num_lab_procedures",
18
+ "num_procedures",
19
+ "num_medications",
20
+ "number_diagnoses",
21
+ "number_inpatient",
22
+ "number_outpatient",
23
+ "number_emergency",
24
+ "race",
25
+ "gender",
26
+ "age",
27
+ "A1Cresult",
28
+ "max_glu_serum",
29
+ "insulin",
30
+ "change",
31
+ "diabetesMed",
32
+ "diag_1",
33
+ "diag_2",
34
+ "diag_3"
35
+ ],
36
+ "numeric_features": [
37
+ "time_in_hospital",
38
+ "num_lab_procedures",
39
+ "num_procedures",
40
+ "num_medications",
41
+ "number_diagnoses",
42
+ "number_inpatient",
43
+ "number_outpatient",
44
+ "number_emergency"
45
+ ],
46
+ "low_cardinality_categorical": [
47
+ "race",
48
+ "gender",
49
+ "age",
50
+ "A1Cresult",
51
+ "max_glu_serum",
52
+ "insulin",
53
+ "change",
54
+ "diabetesMed"
55
+ ],
56
+ "high_cardinality_diagnoses": [
57
+ "diag_1",
58
+ "diag_2",
59
+ "diag_3"
60
+ ],
61
+ "dataset": "UCI Diabetes 130-US Hospitals 1999-2008 (id 296), CC BY 4.0",
62
+ "python_version": "3.14.4",
63
+ "library_versions": {
64
+ "scikit-learn": "1.8.0",
65
+ "numpy": "2.4.6",
66
+ "pandas": "2.3.3",
67
+ "joblib": "1.5.3"
68
+ },
69
+ "training_date": "2026-06-04T20:31:40.072484+00:00"
70
+ }