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
|