Open cesarali opened 1 year ago
Create Config here:
src/graph_bridges/models/backward_rates/backward_rate_config.py
Register Here:
Include In Test Here: tests/backward_rates/test_backward_rates.py
The temporal hollow transformer as a temporal network and config file was defined in:
The experiments for graphs are defined here:
We should try experiments trying different values of:
num_heads=2, num_layers=2, hidden_dim=hidden_dim, ff_hidden_dim=hidden_dim*2, time_embed_dim=128, time_scale_factor=10
as well as:
learning_rate=1e-3, remember to include graphs in :
metrics=["graphs_plots",
"histograms"]
such as obtaining the orca values etc.
I don't know why the architecture is not really working (the simple MLP is working like a charm), I hope is just a matter of finding the right parameters, but maybe is how the time embeddings are included in the architecture or something more fundamental.
We have to fix this!
Carnegie Melon Architecture