yan123yan
commited on
Commit
·
af53761
1
Parent(s):
dfa4f89
fix bugs
Browse files
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 |
|