yehezkielgunawan commited on
Commit
8de32f8
β€’
1 Parent(s): fcb94a3

fix(app.py): :green_heart: use the best model

Browse files
.gitattributes CHANGED
@@ -35,3 +35,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  best_classifier2.pkl filter=lfs diff=lfs merge=lfs -text
37
  best_classifier1.pkl filter=lfs diff=lfs merge=lfs -text
 
 
 
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  best_classifier2.pkl filter=lfs diff=lfs merge=lfs -text
37
  best_classifier1.pkl filter=lfs diff=lfs merge=lfs -text
38
+ best_classifier1_optimized.pkl filter=lfs diff=lfs merge=lfs -text
39
+ best_classifier2_optimized.pkl filter=lfs diff=lfs merge=lfs -text
app.py CHANGED
@@ -16,8 +16,8 @@ priority_score_mapping = {1: "low", 2: "medium", 3: "high"}
16
  problem_domain_mapping = {0: "operational", 1: "tech"}
17
 
18
  # Load the trained Random Forest models
19
- best_classifier1 = joblib.load('best_classifier1.pkl')
20
- best_classifier2 = joblib.load('best_classifier2.pkl')
21
 
22
  # Function to perform predictions
23
  def predict(text):
 
16
  problem_domain_mapping = {0: "operational", 1: "tech"}
17
 
18
  # Load the trained Random Forest models
19
+ best_classifier1 = joblib.load('best_classifier1_optimized.pkl')
20
+ best_classifier2 = joblib.load('best_classifier2_optimized.pkl')
21
 
22
  # Function to perform predictions
23
  def predict(text):
best_classifier1.pkl β†’ best_classifier1_optimized.pkl RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:11cfb0a69900b41beea15edc677a5f24e506100787c97629c54ecc4acf17e152
3
- size 1170993
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a9f001ff46ea8ad35f4f1649b3f8b0fec584d45d3a9a289cc29e6883288f4392
3
+ size 1095297
best_classifier2.pkl β†’ best_classifier2_optimized.pkl RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8dd36d9e99d6b6cf5d04c4f379a107f277cce50ac6584853921327c8548a6fdd
3
- size 404521
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7e0a5134cae8afab54b693de8add405cb027f23f72e3010550ef22103071135b
3
+ size 235145