5f1849fd1c
Added docs to ut part 2
2022-05-27 09:05:14 -04:00
c90efc0403
add check for shm available when num_devices=1 to avoid bus errors
...
-Required modifying tests to reduce memory usage to check doesn't fail during CI
-Also required fixing max_rowsize to be externalized from map+batch() function
2021-09-24 10:24:31 -07:00
2c7d1bf603
Implement new gnn feature
2021-06-07 10:20:06 +08:00
9284b6d6af
Add special case returned value for get feature function
2021-05-18 15:53:01 +08:00
f3895f2b09
Add get edges from nodes function for gnn network
2021-04-10 18:31:13 +08:00
2dc9ba761c
gnn support weight sampling neighbors.
2021-01-28 15:28:44 +08:00
9b3c33e157
minddata iterator output ms_tensor
2020-09-11 11:35:50 +08:00
Lixia Chen
27d5281641
Change epoch count to 1 for python testcases
2020-09-01 11:24:26 -04:00
7a046a1d70
support get_edge_feature
2020-07-10 17:05:23 +08:00
tony_liu2
14899a1410
fix gnn random walk pr 1977 comments
...
add fix to random resize decode crop test case
fix pylint issues
2020-07-07 17:32:08 -04:00
Jonathan Yan
87d2c27c7f
random walk v1
2020-06-15 08:54:10 -04:00
3ece8dd090
1. support get_all_edges, get_nodes_from_edge, get_sampled_neighbors, get_neg_sampled_neighbors and graph_info API
...
2. mod cora and citeseer conversion script
2020-06-10 15:52:26 +08:00
f28f883cac
fix comments error and modify parameter check
2020-05-23 18:31:13 +08:00
34236ce1f1
fix pylint
2020-05-22 20:18:21 +08:00
599a449e0b
Support processing GNN data
2020-05-22 14:15:25 +08:00
David
Lixia Chen
tony_liu2
Jonathan Yan