You are on page 1of 8

demo_regression

August 7, 2022

[1]: from ww_torch import *


INPUT_DIM = 168
OUTPUT_DIM = 9

runner = ww('ww.csv', INPUT_DIM)


plt.rcParams['figure.dpi'] = 300

[2]: runner.run(10)

Fold: 0
Best Val. Loss: 0.524 | On Epoch: 407
Test Loss: 0.554 | On Epoch: 407
Fold: 1
Best Val. Loss: 0.516 | On Epoch: 105
Test Loss: 0.623 | On Epoch: 105
Fold: 2
Best Val. Loss: 0.519 | On Epoch: 129
Test Loss: 0.531 | On Epoch: 129
Fold: 3
Best Val. Loss: 0.457 | On Epoch: 88
Test Loss: 0.309 | On Epoch: 88
Fold: 4
Best Val. Loss: 0.422 | On Epoch: 143
Test Loss: 0.414 | On Epoch: 143
Fold: 5
Best Val. Loss: 0.291 | On Epoch: 151
Test Loss: 0.581 | On Epoch: 151
Fold: 6
Best Val. Loss: 0.635 | On Epoch: 426
Test Loss: 0.559 | On Epoch: 426
Fold: 7
Best Val. Loss: 0.418 | On Epoch: 291
Test Loss: 0.459 | On Epoch: 291
Fold: 8
Best Val. Loss: 0.577 | On Epoch: 10
Test Loss: 0.747 | On Epoch: 10
Fold: 9
Best Val. Loss: 0.510 | On Epoch: 161

1
Test Loss: 0.840 | On Epoch: 161
Validation loss per fold: [0.524 0.516 0.519 0.457 0.422 0.291 0.635 0.418 0.577
0.51 ]
test loss per fold: [0.554 0.623 0.531 0.309 0.414 0.581 0.559 0.459 0.747 0.84
]
Mean loss: 0.487
Test loss: 0.562

[3]: def plot(fold):


mean, std = runner.error_hist(fold)
fig, ax = plt.subplots()
ax.bar(mean.keys(), mean.values(), yerr=std.values(), ecolor='black',␣
↪capsize=10)

ax.set_xticks([1,1.5,2,2.5,3,3.5,4,4.5,5])
ax.set_xlabel('True Label')
ax.set_ylabel('MAE')
ax.set_title(f'Error Histogram per label of fold {fold}')
plt.show()

[4]: for i in range(10):


plot(i)

Evaluating: 0%| | 0/2 [00:00<?, ?it/s]

Evaluating: 0%| | 0/2 [00:00<?, ?it/s]

2
Evaluating: 0%| | 0/2 [00:00<?, ?it/s]

3
Evaluating: 0%| | 0/2 [00:00<?, ?it/s]

Evaluating: 0%| | 0/2 [00:00<?, ?it/s]

4
Evaluating: 0%| | 0/2 [00:00<?, ?it/s]

5
Evaluating: 0%| | 0/2 [00:00<?, ?it/s]

Evaluating: 0%| | 0/2 [00:00<?, ?it/s]

6
Evaluating: 0%| | 0/2 [00:00<?, ?it/s]

7
Evaluating: 0%| | 0/2 [00:00<?, ?it/s]

[ ]:

You might also like