HMPhuoc commited on
Commit
e4b9572
1 Parent(s): 13b8bac

update judgePlus

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -122,10 +122,13 @@ def judgePlus(x):
122
  result_gru = np.round(gru_pred, 2)
123
  result_bert = np.round(bert_pred, 2)
124
  #result_bert = np.round(bert_pred, 2)
125
-
126
- for i in range(6):
127
- result.append((result_lstm[i]+result_gru[i]+result_bert[i])/3)
128
-
 
 
 
129
  return (result)
130
 
131
  def judgeBert(x):
 
122
  result_gru = np.round(gru_pred, 2)
123
  result_bert = np.round(bert_pred, 2)
124
  #result_bert = np.round(bert_pred, 2)
125
+ if((result_lstm[0]+result_gru[0])<(result_bert[0]*2)):
126
+ for i in range(6):
127
+ result.append((result_bert)/1)
128
+ else:
129
+ for i in range(6):
130
+ result.append((result_lstm[i]+result_gru[i])/2)
131
+
132
  return (result)
133
 
134
  def judgeBert(x):