tomonori-masui/split_nodes.py

Last active September 28, 2022 05:08

Star (0) You must be signed in to star a gist
Fork (0) You must be signed in to fork a gist

Select an option

Learn more about clone URLs
Clone this repository at <script src="https://gist.github.com/tomonori-masui/924bf4c664cfe0e6c822cc6d3a6b932f.js"></script>
Save tomonori-masui/924bf4c664cfe0e6c822cc6d3a6b932f to your computer and use it in GitHub Desktop.

Download ZIP

Randomly split nodes in a graph

Raw

split_nodes.py

	import torch_geometric.transforms as T

	split = T.RandomNodeSplit(num_val=0.1, num_test=0.2)
	graph = split(graph)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment