arian81 commited on
Commit
9e3e288
1 Parent(s): 1e71947

init commit

Browse files
Files changed (3) hide show
  1. README.md +13 -0
  2. features.png +0 -0
  3. random_forest_model.joblib +3 -0
README.md CHANGED
@@ -1,3 +1,16 @@
1
  ---
2
  license: mit
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
  ---
4
+
5
+ # Training
6
+
7
+ Random forest model trained on TEDS(TREATMENT EPISODE DATA SET
8
+ ADMISSIONS) dataset. Random search was employed to hyperparamter fine tune the model.
9
+
10
+ # Results
11
+
12
+ An accuracy of 74% was achieved with lots of potential to increase accuracy by fruther cleaning the input data set and feature engineering.
13
+
14
+ ### Most important features
15
+
16
+ ![Most important features](features.png)
features.png ADDED
random_forest_model.joblib ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8eaf166a1d1be8c79a0ebbb4b090bff7562e101e2ba39d03889f73769feb45da
3
+ size 513118921