Add Graph class #403
Add Graph class #403
Conversation
|
@FilippoOlivo I would avoid to write a graph solver, |
FilippoOlivo
force-pushed
the
0.2
branch
10 times, most recently
from
567f207 to
1867274
Compare
dario-coscia
changed the title
FilippoOlivo
force-pushed
the
0.2
branch
5 times, most recently
from
11a43ec to
290641a
Compare
FilippoOlivo
force-pushed
the
0.2
branch
3 times, most recently
from
7ba1b32 to
238f665
Compare
FilippoOlivo
added
pr-to-review
dario-coscia
added
pr-to-fix
FilippoOlivo
force-pushed
the
0.2
branch
2 times, most recently
from
17962fd to
89685a5
Compare
|
@FilippoOlivo @gc031298 This refactoring looks very nice! |
FilippoOlivo
force-pushed
the
0.2
branch
2 times, most recently
from
67d39b0 to
d1058e9
Compare
FilippoOlivo
added
pr-to-review
pina/graph.py
Outdated
| x, pos, edge_index = self._check_input_consistency(x, pos, edge_index) | ||
| print(len(pos)) | ||
| if edge_index is None: |
There was a problem hiding this comment.
i don't know if this is nice to be kept here.. why radius and not knn? Do we really need a temporal graph? One could do a normal RadiusGraph and then add time as additional parameter
There was a problem hiding this comment.
Ok. As for now I remove TemporalGraph
| internal_func=internal_func, | ||
| external_func=external_func) | ||
| self.n_layers = n_layers | ||
| self.forward = self.forward_shared |
There was a problem hiding this comment.
I don't like a lot this forward separation, is there a way to combine the two?
There was a problem hiding this comment.
In order to have an efficient way to store parameters (avoid to use torch.nn.ModuleList with the same model repeated n_layer times), another possible solution is using an if in the forward. Otherwise I can define another 2 classes: one for the shared_weights and one for the non shared_weights. Let me know what how to proceed
There was a problem hiding this comment.
We can keep it like this for the moment, maybe two classes is the best but for a single model I would not care a lot
|
@FilippoOlivo @gc031298 overall very nice, this looks a very nice integration in PINA! Just minor changes, I think we will be able to merge it soon :)) |
…ndling. Remove TemporalGraph class
This PR refers to the issue #400