Skip to content

Commit

Permalink
Update examples/network_traffic_prediction/nr_models.py
Browse files Browse the repository at this point in the history
Co-authored-by: xiang song(charlie.song) <[email protected]>
  • Loading branch information
zhjwy9343 and classicsong authored Dec 13, 2024
1 parent 1876b2b commit ca33686
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/network_traffic_prediction/nr_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -274,7 +274,7 @@ class RgcnNRModel4TS(GSgnnNodeModel):
node_feat_field: dict of list of strings
The feature name list for each node type to be used in the model.
edge_feat_field: dict of list of strings
The feature name list for each node type to be used in the model.
The feature name list for each edge type to be used in the model.
edge_feat_mp_op: str
The operation to be used to compute source node embeddings and edge
embeddings during message passing computation. Options include "concat",
Expand Down

0 comments on commit ca33686

Please sign in to comment.