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 |
|