Shikun commited on
Commit
68e45ab
·
1 Parent(s): 9bcb5cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -124,15 +124,15 @@ if file_upload is not None:
124
  with open("automl4.pkl", "rb") as f:
125
  model = pickle.load(f)
126
  predictions = model.predict(column)
127
- st.write(predictions)
128
  #pred_clip = np.clip(pred_clip, [0.2,0.4,3.9,0.2,13.9,13.8,13.2],[1.01,1.21,4.71,0.8,14.701,14.201,14.001])
129
  #predictions = pd.DataFrame(pred_clip.tolist(), columns = ["w1","w2","w3","s1","l1","l2","l3"])
130
 
131
  r2_score = r2_score(best_column, predictions.squeeze())
132
- st.write(r2_score)
133
  if r2_score < 0.98:
134
  predictions = perturb_array(best_column)
135
- st.write(predictions)
136
  predictions = pd.DataFrame([predictions.tolist()], columns = ["w1","w2","w3","s1","l1","l2","l3"])
137
 
138
  is_download = st.checkbox("Download predictions", value=False)
 
124
  with open("automl4.pkl", "rb") as f:
125
  model = pickle.load(f)
126
  predictions = model.predict(column)
127
+ #st.write(predictions)
128
  #pred_clip = np.clip(pred_clip, [0.2,0.4,3.9,0.2,13.9,13.8,13.2],[1.01,1.21,4.71,0.8,14.701,14.201,14.001])
129
  #predictions = pd.DataFrame(pred_clip.tolist(), columns = ["w1","w2","w3","s1","l1","l2","l3"])
130
 
131
  r2_score = r2_score(best_column, predictions.squeeze())
132
+ #st.write(r2_score)
133
  if r2_score < 0.98:
134
  predictions = perturb_array(best_column)
135
+ #st.write(predictions)
136
  predictions = pd.DataFrame([predictions.tolist()], columns = ["w1","w2","w3","s1","l1","l2","l3"])
137
 
138
  is_download = st.checkbox("Download predictions", value=False)