Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
rLLM documentation

rLLM

INTRODUCTION

  • Graph Data Handle
  • Table Data Handle
  • LLM Data Handle

TUTORIALS

  • Understanding Transform
  • Understanding Convolution
  • Design of GNNs
  • Design of TNNs
  • Design of RTLs
  • Design of LLM Methods
  • Mini-Batch Training

API REFERENCE

  • rllm.data
    • rllm.data.BaseGraph
    • rllm.data.GraphData
    • rllm.data.HeteroGraphData
    • rllm.data.BaseTable
    • rllm.data.TableDataset
    • rllm.data.TableData
    • rllm.data.BaseStorage
    • rllm.data.NodeStorage
    • rllm.data.EdgeStorage
    • rllm.data.MappingView
    • rllm.data.KeysView
    • rllm.data.ValuesView
    • rllm.data.ItemsView
  • rllm.dataloader
    • rllm.dataloader.NeighborLoader
    • rllm.dataloader.BRIDGELoader
    • rllm.dataloader.RelbenchLoader
  • rllm.datasets
    • rllm.datasets.DBLP
    • rllm.datasets.IMDB
    • rllm.datasets.PlanetoidDataset
    • rllm.datasets.TAPEDataset
    • rllm.datasets.TAGDataset
    • rllm.datasets.Titanic
    • rllm.datasets.Adult
    • rllm.datasets.BankMarketing
    • rllm.datasets.ChurnModelling
    • rllm.datasets.TACM12KDataset
    • rllm.datasets.TLF2KDataset
    • rllm.datasets.TML1MDataset
    • rllm.datasets.RelBenchDataset
    • rllm.datasets.RelF1Dataset
    • rllm.datasets.RelBenchTask
    • rllm.datasets.RelBenchTaskType
    • rllm.datasets.RelBenchTableMeta
  • rllm.llm
    • rllm.llm.Predictor
    • rllm.llm.Enhancer
    • rllm.llm.LLMWithRetriever
    • rllm.llm.FeatLLMEngineer
    • rllm.llm.FeatLLM
    • rllm.llm.PromptTemplate
    • rllm.llm.ChatPromptTemplate
    • rllm.llm.LangChainLLM
    • rllm.llm.BaseOutputParser
  • rllm.nn
    • rllm.nn.conv.graph_conv.MessagePassing
    • rllm.nn.conv.graph_conv.GCNConv
    • rllm.nn.conv.graph_conv.LGCConv
    • rllm.nn.conv.graph_conv.GATConv
    • rllm.nn.conv.graph_conv.HANConv
    • rllm.nn.conv.graph_conv.HGTConv
    • rllm.nn.conv.graph_conv.SAGEConv
    • rllm.nn.conv.table_conv.ExcelFormerConv
    • rllm.nn.conv.table_conv.FTTransformerConv
    • rllm.nn.conv.table_conv.SAINTConv
    • rllm.nn.conv.table_conv.TabTransformerConv
    • rllm.nn.conv.table_conv.TromptConv
    • rllm.nn.conv.table_conv.TransTabConv
    • rllm.nn.conv.table_conv.ResNetConv
    • rllm.nn.encoder.FTTransformerPreEncoder
    • rllm.nn.encoder.TabTransformerPreEncoder
    • rllm.nn.encoder.TransTabPreEncoder
    • rllm.nn.encoder.ResNetPreEncoder
    • rllm.nn.encoder.HeteroTemporalEncoder
    • rllm.nn.models.RECT_L
    • rllm.nn.models.BRIDGE
    • rllm.nn.models.TransTab
    • rllm.nn.models.TableResNet
    • rllm.nn.models.HeteroSAGE
    • rllm.nn.models.RDL
    • rllm.nn.models.RelGNN
    • rllm.nn.models.RelGNNModel
    • rllm.nn.loss.ContrastiveLoss
    • rllm.nn.loss.SelfSupervisedVPCL
    • rllm.nn.loss.SupervisedVPCL
  • rllm.preprocessing
    • rllm.preprocessing.df_to_tensor
    • rllm.preprocessing.TokenizerConfig
    • rllm.preprocessing.process_tokenized_column
    • rllm.preprocessing.tokenize_strings
    • rllm.preprocessing.standardize_tokenizer_output
    • rllm.preprocessing.tokenize_merged_cols
    • rllm.preprocessing.save_column_name_tokens
    • rllm.preprocessing.TextEmbedderConfig
    • rllm.preprocessing.embed_text_column
    • rllm.preprocessing.TimestampPreprocessor
    • rllm.preprocessing.FillNAConfig
    • rllm.preprocessing.fillna_by_coltype
  • rllm.transforms
    • rllm.transforms.graph_transforms.NodeTransform
    • rllm.transforms.graph_transforms.EdgeTransform
    • rllm.transforms.graph_transforms.NormalizeFeatures
    • rllm.transforms.graph_transforms.SVDFeatureReduction
    • rllm.transforms.graph_transforms.KNNGraph
    • rllm.transforms.graph_transforms.AddRemainingSelfLoops
    • rllm.transforms.graph_transforms.RemoveSelfLoops
    • rllm.transforms.graph_transforms.GCNNorm
    • rllm.transforms.graph_transforms.GDC
    • rllm.transforms.graph_transforms.GraphTransform
    • rllm.transforms.graph_transforms.GCNTransform
    • rllm.transforms.graph_transforms.RECTTransform
    • rllm.transforms.table_transforms.ColTransform
    • rllm.transforms.table_transforms.ColNormalize
    • rllm.transforms.table_transforms.OneHotTransform
    • rllm.transforms.table_transforms.StackNumerical
    • rllm.transforms.table_transforms.TableTransform
    • rllm.transforms.table_transforms.DefaultTableTransform
    • rllm.transforms.table_transforms.TabTransformerTransform
    • rllm.transforms.utils.RemoveTrainingClasses
  • rllm.utils
    • rllm.utils.download_url
    • rllm.utils.download_google_url
    • rllm.utils.extract_zip
    • rllm.utils.sparse_mx_to_torch_sparse_tensor
    • rllm.utils.is_torch_sparse_tensor
    • rllm.utils.get_indices
    • rllm.utils.is_undirected
    • rllm.utils.to_undirected
    • rllm.utils.adj_to_edge_index
    • rllm.utils.seg_sum
    • rllm.utils.seg_softmax
    • rllm.utils.sort_edge_index
    • rllm.utils.index_to_ptr
    • rllm.utils.lexsort
    • rllm.utils.remap_keys
    • rllm.utils.CastMixin

CHEATSHEETS

  • Model Cheatsheet

EXTERNAL RESOURCES

  • External Resources
Back to top

rllm.datasets.RelBenchTask¶

class rllm.datasets.RelBenchTask(task_name: str, task_type: rllm.datasets.relbench.base.RelBenchTaskType, entity_col: str, entity_table: str, time_col: str, target_col: str, timedelta: pandas._libs.tslibs.timedeltas.Timedelta, num_eval_timestamps: int, task_data_dict: Dict[str, Tuple[pandas.core.frame.DataFrame, rllm.datasets.relbench.base.RelBenchTableMeta]])[source]¶

Bases: object

Next
rllm.datasets.RelBenchTaskType
Previous
rllm.datasets.RelF1Dataset
Copyright © 2024, rLLM Team
Made with Sphinx and @pradyunsg's Furo
On this page
  • rllm.datasets.RelBenchTask
    • RelBenchTask