deva8217 commited on
Commit
5d2f223
·
verified ·
1 Parent(s): f52a53a

Register best model + threshold

Browse files
Files changed (3) hide show
  1. README.md +3 -3
  2. best_model.joblib +2 -2
  3. threshold.json +1 -1
README.md CHANGED
@@ -1,5 +1,5 @@
1
  # Wellness Tourism Purchase Model
2
  Best model: **RandomForest**
3
- Validation F1: **0.7403**
4
- Test metrics: {"roc_auc": 0.93672031908574, "auprc": 0.8173926013577065, "accuracy": 0.8680387409200968, "precision": 0.6157407407407407, "recall": 0.8364779874213837, "f1": 0.7093333333333334}
5
- Threshold: **0.24**
 
1
  # Wellness Tourism Purchase Model
2
  Best model: **RandomForest**
3
+ Validation F1: **0.7429**
4
+ Test metrics: {"roc_auc": 0.9482475743260445, "auprc": 0.8453597025735655, "accuracy": 0.9067796610169492, "precision": 0.7440476190476191, "recall": 0.7861635220125787, "f1": 0.764525993883792}
5
+ Threshold: **0.32**
best_model.joblib CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c316f2975b06da59e457ab1f1d4a9b7fe63a3b98d6c45524876028e61d2e6776
3
- size 4553227
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7a5e502a5a2676dda0adf7195679b55899e9625f2fe57246bcb9b7df52912bcd
3
+ size 5927147
threshold.json CHANGED
@@ -1 +1 @@
1
- {"threshold": 0.24000000000000002}
 
1
+ {"threshold": 0.32}