yan123yan commited on
Commit
af53761
·
1 Parent(s): dfa4f89
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -68,6 +68,8 @@ lstm_cpu_start_time = time.time()
68
  with torch.no_grad():
69
  lstm_cpu_preds = lstm_cpu_model(input_data[:, 100:300, :])
70
  lstm_cpu_end_time = time.time()
 
 
71
 
72
  del lstm_cpu_model
73
 
@@ -91,6 +93,8 @@ with torch.no_grad():
91
  y_hat = torch.concatenate([y_hat, output], dim=1)
92
  tcn_cpu_preds = y_hat
93
  tcn_cpu_end_time = time.time()
 
 
94
 
95
  del tcn_cpu_model
96
 
 
68
  with torch.no_grad():
69
  lstm_cpu_preds = lstm_cpu_model(input_data[:, 100:300, :])
70
  lstm_cpu_end_time = time.time()
71
+ lstm_innv_preds = scaler.inverse_transform(lstm_cpu_preds.squeeze().cpu().numpy())
72
+ lstm_normal_preds = lstm_cpu_preds.squeeze().cpu().numpy()
73
 
74
  del lstm_cpu_model
75
 
 
93
  y_hat = torch.concatenate([y_hat, output], dim=1)
94
  tcn_cpu_preds = y_hat
95
  tcn_cpu_end_time = time.time()
96
+ tcn_innv_preds = scaler.inverse_transform(tcn_cpu_preds.squeeze().cpu().numpy())
97
+ tcn_normal_preds = tcn_cpu_preds.squeeze().cpu().numpy()
98
 
99
  del tcn_cpu_model
100