Isabel Gwara commited on
Commit
7b32eb4
·
1 Parent(s): f0f767a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
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
- global acc
102
- acc = metrics.accuracy_score(y_test, y_pred)
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", str(round(acc * 100, 1)) + '%')
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