Spaces:
Runtime error
Runtime error
Isabel Gwara
commited on
Commit
·
7b32eb4
1
Parent(s):
f0f767a
Update app.py
Browse files
app.py
CHANGED
|
@@ -98,9 +98,10 @@ def train_model():
|
|
| 98 |
with open('model.pkl', 'wb') as f:
|
| 99 |
pkl.dump(model, f)
|
| 100 |
|
| 101 |
-
|
| 102 |
-
|
| 103 |
-
|
|
|
|
| 104 |
return model
|
| 105 |
|
| 106 |
### -------------------------------- ###
|
|
@@ -110,6 +111,8 @@ def train_model():
|
|
| 110 |
try:
|
| 111 |
with open('model.pkl', 'rb') as f:
|
| 112 |
model = pkl.load(f)
|
|
|
|
|
|
|
| 113 |
except FileNotFoundError as e:
|
| 114 |
model = train_model()
|
| 115 |
|
|
@@ -169,7 +172,7 @@ if form.form_submit_button("Submit to get your recommendation!"):
|
|
| 169 |
|
| 170 |
col1, col2 = st.columns(2)
|
| 171 |
col1.metric("Number of Options", len(cat_value_dicts[final_colname]))
|
| 172 |
-
col2.metric("Model Accuracy",
|
| 173 |
# st.metric("Most Important Question", get_feat())
|
| 174 |
|
| 175 |
|
|
|
|
| 98 |
with open('model.pkl', 'wb') as f:
|
| 99 |
pkl.dump(model, f)
|
| 100 |
|
| 101 |
+
with open('acc.txt', 'w+') as f:
|
| 102 |
+
acc = metrics.accuracy_score(y_test, y_pred)
|
| 103 |
+
f.write(str(round(acc * 100, 1)) + '%')
|
| 104 |
+
|
| 105 |
return model
|
| 106 |
|
| 107 |
### -------------------------------- ###
|
|
|
|
| 111 |
try:
|
| 112 |
with open('model.pkl', 'rb') as f:
|
| 113 |
model = pkl.load(f)
|
| 114 |
+
with open('acc.txt', 'r') as f:
|
| 115 |
+
acc = f.read()
|
| 116 |
except FileNotFoundError as e:
|
| 117 |
model = train_model()
|
| 118 |
|
|
|
|
| 172 |
|
| 173 |
col1, col2 = st.columns(2)
|
| 174 |
col1.metric("Number of Options", len(cat_value_dicts[final_colname]))
|
| 175 |
+
col2.metric("Model Accuracy", + '%')
|
| 176 |
# st.metric("Most Important Question", get_feat())
|
| 177 |
|
| 178 |
|