HMPhuoc commited on
Commit
dab8507
1 Parent(s): 338b896

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,7 +25,7 @@ def LSTM_predict(x):
25
  t1 = time.time()
26
  pred_proba = LSTM_model.predict([x])[0]
27
  t2 = time.time()
28
- print(f'{t2-t1}s')
29
 
30
  pred_proba = [round(i,2) for i in pred_proba]
31
 
@@ -35,7 +35,7 @@ def GRU_predict(x):
35
  t1 = time.time()
36
  pred_proba = GRU_model.predict([x])[0]
37
  t2 = time.time()
38
- print(f'{t2-t1}s')
39
 
40
  pred_proba = [round(i,2) for i in pred_proba]
41
 
 
25
  t1 = time.time()
26
  pred_proba = LSTM_model.predict([x])[0]
27
  t2 = time.time()
28
+ print(f'LSTM: {t2-t1}s')
29
 
30
  pred_proba = [round(i,2) for i in pred_proba]
31
 
 
35
  t1 = time.time()
36
  pred_proba = GRU_model.predict([x])[0]
37
  t2 = time.time()
38
+ print(f'GRU: {t2-t1}s')
39
 
40
  pred_proba = [round(i,2) for i in pred_proba]
41