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('
|
20 |
-
best_classifier2 = joblib.load('
|
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:
|
3 |
-
size
|
|
|
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:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7e0a5134cae8afab54b693de8add405cb027f23f72e3010550ef22103071135b
|
3 |
+
size 235145
|