Commit 6b111617 authored by Marco Cristoforetti's avatar Marco Cristoforetti
Browse files

classifier

parent 8453acfc
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
No preview for this file type
......@@ -220,16 +220,16 @@ loss_f = nn.L1Loss()
loss_fc= nn.CrossEntropyLoss(weight = weights_c)
nhidden_i = 2
nhidden_o = 96
n_out_i = 8
nhidden_o = 48
n_out_i = 4
before = BEFORE
nvars = data_in_scaled.shape[-1]
dst_net = DSTnet(nvars, nhidden_i, nhidden_o, n_out_i, before, AFTER).to(device)
print(dst_net)
num_epochs = 10000
lr = 1e-5
num_epochs = 5000
lr = 1e-4
optimizer = torch.optim.Adam(dst_net.parameters(), lr=lr)#, weight_decay=1e-5)
history_tr = np.zeros(num_epochs)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment