grid_search issue about pycox HOT 2 CLOSED

Hi! This is a good question, and sorry for not coming back to you sooner.

To me is looks like the net is reinitialised fine, but I can see that you use the same set of callbacks for every iteration. I don't know what you callback contains, but I assume it might be an EarlyStopping object. If this is the case, you need to reinitialise this too, meaning you end up with something like this

for lr in [0.01, 0.1, 0.001]:
    net = tt.practical.MLPVanilla(in_features, num_nodes, out_features, batch_norm,
                              dropout, output_bias=output_bias)
    model = CoxPH(net, tt.optim.Adam)
    model.optimizer.set_lr(0.01)
    callbacks = [tt.callbacks.EarlyStopping()]
    log = model.fit(x_train, y_train, batch_size, epochs, callbacks, verbose=True,
                    val_data=val, val_batch_size=batch_size)

Does this solve the issue?

from pycox.

Yes, it works for the issue!
thank you.

from pycox.