Loaded data of shape (253, 12)
	x: (253, 8)
	y: (253, 4)

Initialized dataloaders
	train: 7 batches (202 samples)
	test: 2 batches (51 samples)

Initialized model
NeuralNetwork(
  (features): Sequential(
    (0): Linear(in_features=8, out_features=512, bias=True)
    (1): ReLU()
    (2): BatchNorm1d(512, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    (3): Dropout(p=0.4, inplace=False)
    (4): Linear(in_features=512, out_features=256, bias=True)
    (5): ReLU()
    (6): BatchNorm1d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    (7): Dropout(p=0.4, inplace=False)
    (8): Linear(in_features=256, out_features=128, bias=True)
    (9): ReLU()
    (10): BatchNorm1d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    (11): Dropout(p=0.4, inplace=False)
    (12): Linear(in_features=128, out_features=64, bias=True)
    (13): ReLU()
    (14): BatchNorm1d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True)
    (15): Dropout(p=0.4, inplace=False)
    (16): Linear(in_features=64, out_features=4, bias=True)
  )
)

Epoch 1/200...
Epoch 1/200
Train loss: 4.0131
Test loss: 3.9398

Epoch 2/200...
Epoch 2/200
Train loss: 3.9663
Test loss: 4.2228

Epoch 3/200...
Epoch 3/200
Train loss: 3.8031
Test loss: 4.1961

Epoch 4/200...
Epoch 4/200
Train loss: 3.8808
Test loss: 4.0986

Epoch 5/200...
Epoch 5/200
Train loss: 3.6650
Test loss: 4.1003

Epoch 6/200...
Epoch 6/200
Train loss: 3.7450
Test loss: 3.9062

Epoch 7/200...
Epoch 7/200
Train loss: 3.5706
Test loss: 3.5473

Epoch 8/200...
Epoch 8/200
Train loss: 3.5009
Test loss: 3.5034

Epoch 9/200...
Epoch 9/200
Train loss: 3.1997
Test loss: 3.3960

Epoch 10/200...
Epoch 10/200
Train loss: 3.1823
Test loss: 3.1974

Epoch 11/200...
Epoch 11/200
Train loss: 3.0005
Test loss: 3.0767

Epoch 12/200...
Epoch 12/200
Train loss: 3.0974
Test loss: 2.9992

Epoch 13/200...
Epoch 13/200
Train loss: 3.0595
Test loss: 2.9776

Epoch 14/200...
Epoch 14/200
Train loss: 2.9526
Test loss: 2.8866

Epoch 15/200...
Epoch 15/200
Train loss: 2.7826
Test loss: 2.8075

Epoch 16/200...
Epoch 16/200
Train loss: 2.5710
Test loss: 2.7168

Epoch 17/200...
Epoch 17/200
Train loss: 2.5372
Test loss: 2.7869

Epoch 18/200...
Epoch 18/200
Train loss: 2.4086
Test loss: 2.5646

Epoch 19/200...
Epoch 19/200
Train loss: 2.3438
Test loss: 2.4339

Epoch 20/200...
Epoch 20/200
Train loss: 2.2816
Test loss: 2.2510

Epoch 21/200...
Epoch 21/200
Train loss: 2.1555
Test loss: 2.0716

Epoch 22/200...
Epoch 22/200
Train loss: 2.0383
Test loss: 1.9634

Epoch 23/200...
Epoch 23/200
Train loss: 1.8637
Test loss: 1.8379

Epoch 24/200...
Epoch 24/200
Train loss: 1.7369
Test loss: 1.6801

Epoch 25/200...
Epoch 25/200
Train loss: 1.6588
Test loss: 1.7402

Epoch 26/200...
Epoch 26/200
Train loss: 1.5274
Test loss: 1.5483

Epoch 27/200...
Epoch 27/200
Train loss: 1.4587
Test loss: 1.3686

Epoch 28/200...
Epoch 28/200
Train loss: 1.3350
Test loss: 1.1606

Epoch 29/200...
Epoch 29/200
Train loss: 1.2166
Test loss: 1.0983

Epoch 30/200...
Epoch 30/200
Train loss: 1.0755
Test loss: 1.0600

Epoch 31/200...
Epoch 31/200
Train loss: 1.0913
Test loss: 1.0042

Epoch 32/200...
Epoch 32/200
Train loss: 1.1916
Test loss: 0.9510

Epoch 33/200...
Epoch 33/200
Train loss: 1.0838
Test loss: 0.8147

Epoch 34/200...
Epoch 34/200
Train loss: 0.8440
Test loss: 0.7637

Epoch 35/200...
Epoch 35/200
Train loss: 1.0334
Test loss: 0.7435

Epoch 36/200...
Epoch 36/200
Train loss: 1.0374
Test loss: 0.7000

Epoch 37/200...
Epoch 37/200
Train loss: 0.9321
Test loss: 0.7876

Epoch 38/200...
Epoch 38/200
Train loss: 0.8030
Test loss: 1.1057

Epoch 39/200...
Epoch 39/200
Train loss: 1.0274
Test loss: 1.2015

Epoch 40/200...
Epoch 40/200
Train loss: 0.9833
Test loss: 0.9574

Epoch 41/200...
Epoch 41/200
Train loss: 0.7686
Test loss: 1.1230

Epoch 42/200...
Epoch 42/200
Train loss: 0.9205
Test loss: 0.7713

Epoch 43/200...
Epoch 43/200
Train loss: 0.8096
Test loss: 0.6233

Epoch 44/200...
Epoch 44/200
Train loss: 0.7608
Test loss: 0.5627

Epoch 45/200...
Epoch 45/200
Train loss: 0.8330
Test loss: 0.5492

Epoch 46/200...
Epoch 46/200
Train loss: 0.9530
Test loss: 0.5391

Epoch 47/200...
Epoch 47/200
Train loss: 0.7694
Test loss: 0.5384

Epoch 48/200...
Epoch 48/200
Train loss: 0.7705
Test loss: 0.5606

Epoch 49/200...
Epoch 49/200
Train loss: 0.8378
Test loss: 0.6662

Epoch 50/200...
Epoch 50/200
Train loss: 0.8091
Test loss: 0.6342

Epoch 51/200...
Epoch 51/200
Train loss: 0.9266
Test loss: 0.6289

Epoch 52/200...
Epoch 52/200
Train loss: 0.6361
Test loss: 0.6519

Epoch 53/200...
Epoch 53/200
Train loss: 0.7711
Test loss: 0.5803

Epoch 54/200...
Epoch 54/200
Train loss: 0.7270
Test loss: 0.5417

Epoch 55/200...
Epoch 55/200
Train loss: 0.7493
Test loss: 0.5667

Epoch 56/200...
Epoch 56/200
Train loss: 0.7027
Test loss: 0.5813

Epoch 57/200...
Epoch 57/200
Train loss: 0.6886
Test loss: 0.5926

Epoch 58/200...
Epoch 58/200
Train loss: 0.7474
Test loss: 0.6215

Epoch 59/200...
Epoch 59/200
Train loss: 0.7878
Test loss: 0.5791

Epoch 60/200...
Epoch 60/200
Train loss: 0.7738
Test loss: 0.5804

Epoch 61/200...
Epoch 61/200
Train loss: 0.7754
Test loss: 0.5813

Epoch 62/200...
Epoch 62/200
Train loss: 0.7724
Test loss: 0.6502

Epoch 63/200...
Epoch 63/200
Train loss: 0.6556
Test loss: 0.6145

Epoch 64/200...
Epoch 64/200
Train loss: 0.7091
Test loss: 0.6644

Epoch 65/200...
Epoch 65/200
Train loss: 0.7002
Test loss: 0.7436

Epoch 66/200...
Epoch 66/200
Train loss: 0.7678
Test loss: 0.6759

Epoch 67/200...
Epoch 67/200
Train loss: 0.7660
Test loss: 0.6460

Epoch 68/200...
Epoch 68/200
Train loss: 0.7112
Test loss: 0.6855

Epoch 69/200...
Epoch 69/200
Train loss: 0.7324
Test loss: 0.6257

Epoch 70/200...
Epoch 70/200
Train loss: 0.7896
Test loss: 0.6159

Epoch 71/200...
Epoch 71/200
Train loss: 0.6911
Test loss: 0.6996

Epoch 72/200...
Epoch 72/200
Train loss: 0.6484
Test loss: 0.7175

Epoch 73/200...
Epoch 73/200
Train loss: 0.7880
Test loss: 0.6342

Epoch 74/200...
Epoch 74/200
Train loss: 0.6760
Test loss: 0.6688

Epoch 75/200...
Epoch 75/200
Train loss: 0.8485
Test loss: 0.6617

Epoch 76/200...
Epoch 76/200
Train loss: 0.8207
Test loss: 0.7157

Epoch 77/200...
Epoch 77/200
Train loss: 0.7120
Test loss: 0.7030

Epoch 78/200...
Epoch 78/200
Train loss: 0.7311
Test loss: 0.6930

Epoch 79/200...
Epoch 79/200
Train loss: 0.7560
Test loss: 0.6722

Epoch 80/200...
Epoch 80/200
Train loss: 0.7750
Test loss: 0.7678

Epoch 81/200...
Epoch 81/200
Train loss: 0.6823
Test loss: 0.7445

Epoch 82/200...
Epoch 82/200
Train loss: 0.8072
Test loss: 0.7513

Epoch 83/200...
Epoch 83/200
Train loss: 0.7767
Test loss: 0.6822

Epoch 84/200...
Epoch 84/200
Train loss: 0.7502
Test loss: 0.6625

Epoch 85/200...
Epoch 85/200
Train loss: 0.7494
Test loss: 0.7062

Epoch 86/200...
Epoch 86/200
Train loss: 0.6989
Test loss: 0.6885

Epoch 87/200...
Epoch 87/200
Train loss: 0.7874
Test loss: 0.6544

Epoch 88/200...
Epoch 88/200
Train loss: 0.7139
Test loss: 0.6900

Epoch 89/200...
Epoch 89/200
Train loss: 0.8022
Test loss: 0.6870

Epoch 90/200...
Epoch 90/200
Train loss: 0.6817
Test loss: 0.7228

Epoch 91/200...
Epoch 91/200
Train loss: 0.7186
Test loss: 0.6956

Epoch 92/200...
Epoch 92/200
Train loss: 0.7708
Test loss: 0.6585

Epoch 93/200...
Epoch 93/200
Train loss: 0.7071
Test loss: 0.7329

Epoch 94/200...
Epoch 94/200
Train loss: 0.8012
Test loss: 0.7796

Epoch 95/200...
Epoch 95/200
Train loss: 0.7187
Test loss: 0.6991

Epoch 96/200...
Epoch 96/200
Train loss: 0.7477
Test loss: 0.5824

Epoch 97/200...
Epoch 97/200
Train loss: 0.7217
Test loss: 0.6559

Epoch 98/200...
Epoch 98/200
Train loss: 0.7241
Test loss: 0.6721

Epoch 99/200...
Epoch 99/200
Train loss: 0.7947
Test loss: 0.6389

Epoch 100/200...
Epoch 100/200
Train loss: 0.8491
Test loss: 0.7180

Epoch 101/200...
Epoch 101/200
Train loss: 0.7193
Test loss: 0.7870

Epoch 102/200...
Epoch 102/200
Train loss: 0.7741
Test loss: 0.7797

Epoch 103/200...
Epoch 103/200
Train loss: 0.6816
Test loss: 0.7478

Epoch 104/200...
Epoch 104/200
Train loss: 0.7723
Test loss: 0.7173

Epoch 105/200...
Epoch 105/200
Train loss: 0.8082
Test loss: 0.7357

Epoch 106/200...
Epoch 106/200
Train loss: 0.7489
Test loss: 0.7244

Epoch 107/200...
Epoch 107/200
Train loss: 0.6754
Test loss: 0.7175

Epoch 108/200...
Epoch 108/200
Train loss: 0.6513
Test loss: 0.7161

Epoch 109/200...
Epoch 109/200
Train loss: 0.7596
Test loss: 0.6349

Epoch 110/200...
Epoch 110/200
Train loss: 0.7473
Test loss: 0.6376

Epoch 111/200...
Epoch 111/200
Train loss: 0.7011
Test loss: 0.6680

Epoch 112/200...
Epoch 112/200
Train loss: 0.6889
Test loss: 0.6760

Epoch 113/200...
Epoch 113/200
Train loss: 0.8321
Test loss: 0.7368

Epoch 114/200...
Epoch 114/200
Train loss: 0.6969
Test loss: 0.7440

Epoch 115/200...
Epoch 115/200
Train loss: 0.7902
Test loss: 0.6686

Epoch 116/200...
Epoch 116/200
Train loss: 0.6553
Test loss: 0.7150

Epoch 117/200...
Epoch 117/200
Train loss: 0.8302
Test loss: 0.6776

Epoch 118/200...
Epoch 118/200
Train loss: 0.7135
Test loss: 0.6969

Epoch 119/200...
Epoch 119/200
Train loss: 0.7519
Test loss: 0.7012

Epoch 120/200...
Epoch 120/200
Train loss: 0.8335
Test loss: 0.7362

Epoch 121/200...
Epoch 121/200
Train loss: 0.8850
Test loss: 0.8066

Epoch 122/200...
Epoch 122/200
Train loss: 0.7743
Test loss: 0.7470

Epoch 123/200...
Epoch 123/200
Train loss: 0.7290
Test loss: 0.7082

Epoch 124/200...
Epoch 124/200
Train loss: 0.6814
Test loss: 0.6955

Epoch 125/200...
Epoch 125/200
Train loss: 0.6713
Test loss: 0.5742

Epoch 126/200...
Epoch 126/200
Train loss: 0.6581
Test loss: 0.6084

Epoch 127/200...
Epoch 127/200
Train loss: 0.8018
Test loss: 0.6758

Epoch 128/200...
Epoch 128/200
Train loss: 0.7031
Test loss: 0.6852

Epoch 129/200...
Epoch 129/200
Train loss: 0.7510
Test loss: 0.6690

Epoch 130/200...
Epoch 130/200
Train loss: 0.7069
Test loss: 0.6762

Epoch 131/200...
Epoch 131/200
Train loss: 0.8548
Test loss: 0.6829

Epoch 132/200...
Epoch 132/200
Train loss: 0.7175
Test loss: 0.6330

Epoch 133/200...
Epoch 133/200
Train loss: 0.7013
Test loss: 0.6705

Epoch 134/200...
Epoch 134/200
Train loss: 0.7324
Test loss: 0.6898

Epoch 135/200...
Epoch 135/200
Train loss: 0.7172
Test loss: 0.6729

Epoch 136/200...
Epoch 136/200
Train loss: 0.7561
Test loss: 0.5759

Epoch 137/200...
Epoch 137/200
Train loss: 0.7026
Test loss: 0.5536

Epoch 138/200...
Epoch 138/200
Train loss: 0.7878
Test loss: 0.6006

Epoch 139/200...
Epoch 139/200
Train loss: 0.6797
Test loss: 0.6154

Epoch 140/200...
Epoch 140/200
Train loss: 0.7446
Test loss: 0.6897

Epoch 141/200...
Epoch 141/200
Train loss: 0.8153
Test loss: 0.7276

Epoch 142/200...
Epoch 142/200
Train loss: 0.6595
Test loss: 0.6968

Epoch 143/200...
Epoch 143/200
Train loss: 0.8059
Test loss: 0.6889

Epoch 144/200...
Epoch 144/200
Train loss: 0.8237
Test loss: 0.6829

Epoch 145/200...
Epoch 145/200
Train loss: 0.7213
Test loss: 0.6691

Epoch 146/200...
Epoch 146/200
Train loss: 0.7734
Test loss: 0.6631

Epoch 147/200...
Epoch 147/200
Train loss: 0.6768
Test loss: 0.6656

Epoch 148/200...
Epoch 148/200
Train loss: 0.6597
Test loss: 0.6877

Epoch 149/200...
Epoch 149/200
Train loss: 0.6632
Test loss: 0.7286

Epoch 150/200...
Epoch 150/200
Train loss: 0.7452
Test loss: 0.7839

Epoch 151/200...
Epoch 151/200
Train loss: 0.7763
Test loss: 0.7433

Epoch 152/200...
Epoch 152/200
Train loss: 0.6407
Test loss: 0.7554

Epoch 153/200...
Epoch 153/200
Train loss: 0.7199
Test loss: 0.7148

Epoch 154/200...
Epoch 154/200
Train loss: 0.8178
Test loss: 0.6977

Epoch 155/200...
Epoch 155/200
Train loss: 0.7121
Test loss: 0.7091

Epoch 156/200...
Epoch 156/200
Train loss: 0.6474
Test loss: 0.7304

Epoch 157/200...
Epoch 157/200
Train loss: 0.7057
Test loss: 0.7056

Epoch 158/200...
Epoch 158/200
Train loss: 0.7195
Test loss: 0.7261

Epoch 159/200...
Epoch 159/200
Train loss: 0.7129
Test loss: 0.7462

Epoch 160/200...
Epoch 160/200
Train loss: 0.6942
Test loss: 0.7279

Epoch 161/200...
Epoch 161/200
Train loss: 0.7624
Test loss: 0.6452

Epoch 162/200...
Epoch 162/200
Train loss: 0.7654
Test loss: 0.7049

Epoch 163/200...
Epoch 163/200
Train loss: 0.7225
Test loss: 0.7088

Epoch 164/200...
Epoch 164/200
Train loss: 0.7176
Test loss: 0.7594

Epoch 165/200...
Epoch 165/200
Train loss: 0.8333
Test loss: 0.7366

Epoch 166/200...
Epoch 166/200
Train loss: 0.7044
Test loss: 0.6740

Epoch 167/200...
Epoch 167/200
Train loss: 0.8308
Test loss: 0.6772

Epoch 168/200...
Epoch 168/200
Train loss: 0.6969
Test loss: 0.6837

Epoch 169/200...
Epoch 169/200
Train loss: 0.6474
Test loss: 0.6878

Epoch 170/200...
Epoch 170/200
Train loss: 0.6399
Test loss: 0.6449

Epoch 171/200...
Epoch 171/200
Train loss: 0.7689
Test loss: 0.7162

Epoch 172/200...
Epoch 172/200
Train loss: 0.7038
Test loss: 0.7270

Epoch 173/200...
Epoch 173/200
Train loss: 0.7600
Test loss: 0.6172

Epoch 174/200...
Epoch 174/200
Train loss: 0.7420
Test loss: 0.6078

Epoch 175/200...
Epoch 175/200
Train loss: 0.8188
Test loss: 0.6560

Epoch 176/200...
Epoch 176/200
Train loss: 0.7025
Test loss: 0.7334

Epoch 177/200...
Epoch 177/200
Train loss: 0.7360
Test loss: 0.6803

Epoch 178/200...
Epoch 178/200
Train loss: 0.6800
Test loss: 0.7532

Epoch 179/200...
Epoch 179/200
Train loss: 0.7361
Test loss: 0.7004

Epoch 180/200...
Epoch 180/200
Train loss: 0.6987
Test loss: 0.7876

Epoch 181/200...
Epoch 181/200
Train loss: 0.7724
Test loss: 0.7623

Epoch 182/200...
Epoch 182/200
Train loss: 0.7139
Test loss: 0.6503

Epoch 183/200...
Epoch 183/200
Train loss: 0.7242
Test loss: 0.6773

Epoch 184/200...
Epoch 184/200
Train loss: 0.6981
Test loss: 0.6663

Epoch 185/200...
Epoch 185/200
Train loss: 0.6922
Test loss: 0.6833

Epoch 186/200...
Epoch 186/200
Train loss: 0.6562
Test loss: 0.6706

Epoch 187/200...
Epoch 187/200
Train loss: 0.7907
Test loss: 0.7122

Epoch 188/200...
Epoch 188/200
Train loss: 0.7206
Test loss: 0.6734

Epoch 189/200...
Epoch 189/200
Train loss: 0.7055
Test loss: 0.7764

Epoch 190/200...
Epoch 190/200
Train loss: 0.7314
Test loss: 0.6786

Epoch 191/200...
Epoch 191/200
Train loss: 0.7868
Test loss: 0.6134

Epoch 192/200...
Epoch 192/200
Train loss: 0.7750
Test loss: 0.6292

Epoch 193/200...
Epoch 193/200
Train loss: 0.7063
Test loss: 0.6472

Epoch 194/200...
Epoch 194/200
Train loss: 0.7661
Test loss: 0.6108

Epoch 195/200...
Epoch 195/200
Train loss: 0.7580
Test loss: 0.6831

Epoch 196/200...
Epoch 196/200
Train loss: 0.7855
Test loss: 0.6626

Epoch 197/200...
Epoch 197/200
Train loss: 0.8386
Test loss: 0.6444

Epoch 198/200...
Epoch 198/200
Train loss: 0.8165
Test loss: 0.5930

Epoch 199/200...
Epoch 199/200
Train loss: 0.6654
Test loss: 0.6593

Epoch 200/200...
Epoch 200/200
Train loss: 0.7700
Test loss: 0.7331

Average train loss: 1.0442
Average test loss: 0.9947
Model saved to model.pth

Evaluation Metrics:
Accuracy:  0.7059
Precision: 0.6176
Recall:    0.7059
F1 Score:  0.5995
