<jupyter_start><jupyter_text>The following function allows you to train the provided models with a feature-augmented version of the datasets.     zr.perform_collab_prediction(Dataset, Model, random_seed)The available options for the dataset are: "OGBL-COLLAB", "OGBL-COLLAB_3Cl", "OGBL-COLLAB_4Cl","OGBL-COLLAB_5Cl", "OGBL-COLLAB_34Cl", "OGBL-COLLAB_345Cl"The available models are "GAT", "GCN", "MoNet", "GraphSage" and "GatedGCN_E_PE".A random seed of choise needs to be chosen as well (the results in the paper are obtained by averaging over 41, 42, 43 and 44).An example (to reproduce the best results for the  model) is provided below:<jupyter_code>cp.perform_collab_prediction("OGBL-COLLAB_5Cl", "GCN", 43)<jupyter_output><empty_output>
{'fraction_non_alphanumeric': 0.09851551956815115, 'fraction_numerical': 0.02564102564102564, 'mean_word_length': 6.649484536082475, 'pattern_counts': {'":': 0, '<': 5, '<?xml version=': 0, '>': 5, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'chain_length': '1', 'dir': 'jupyter-structured-clean-dedup', 'id': '494585', 'license': 'MIT', 'n_tokens_mistral': 275, 'n_tokens_neox': 250, 'n_words': 79, 'path': 'COLLAB_Example.ipynb', 'repo_name': 'MrRyschkov/LGP-GNN', 'size': '741'}