ggfm.data.get_train_val_test_split¶

class ggfm.data.get_train_val_test_split(graph, train_ratio, val_ratio)[source]¶

Bases:

Split the dataset into train, validation, and test sets.

Parameters:

graph – The graph to split.
train_ratio (float) – The proportion of the dataset to include in the train split.
val_ratio (float) – The proportion of the dataset to include in the validation split.

Return type:

tuple of tensor