Add Splitting Method
To add a new edge splitting method follow these steps:
Fork the dataset, pull the latest changes and create a new branch add-splitting-{split_name}
In the
framework/dataloader/edge_splitter/edge_splitter.pyimplement a function that performs the splitting.This function must return a list of edges of type
t.List[t.Tuple[UserNode, ItemNode]]
In the same file, add the splitting method name to the list self.supported_methods.
Add the splitting method into
split().Add the method to the switch case.
Make sure to check if the parameters are valid. Otherwise, a
ValueErrormust be raised.
Make a Pull Request on Github.