Index

A

add_module() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

aggregate() (tart.representation.models.WeightedGINConv method) apply() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

B

BasicGNN (class in tart.representation.models) bfloat16() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) buffers() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

build_feature_configs() (in module tart.representation.config) build_model() (in module tart.utils.model_utils) build_model_configs() (in module tart.representation.config) build_optimizer() (in module tart.utils.model_utils) build_optimizer_configs() (in module tart.representation.config)

C

children() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) codebert_bpe_encoder() (in module tart.representation.encoders) codebert_encoder() (in module tart.representation.encoders) Corpus (class in tart.representation.dataset)

cpu() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) criterion() (tart.representation.models.SubgraphEmbedder method) cuda() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

D

double() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

download() (tart.representation.dataset.GraphDataset method)

E

edge_update() (tart.representation.models.WeightedGINConv method) edge_updater() (tart.representation.models.WeightedGINConv method) embed_main() (in module tart.inference.embed) eval() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

extra_repr() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

F

featurize_graph() (in module tart.utils.graph_utils) float() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

forward() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method)

G

gen_data_loader() (tart.representation.dataset.Corpus method) generate_embeddings() (in module tart.inference.embed) generate_neighborhoods() (in module tart.inference.embed) get() (tart.representation.dataset.GraphDataset method) get_buffer() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) get_device() (in module tart.utils.model_utils) get_extra_state() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

get_feature_encoder() (in module tart.representation.encoders) get_neighborhoods() (in module tart.inference.embed) get_parameter() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) get_submodule() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) get_summary() (tart.representation.dataset.GraphDataset method) get_torch_tensor_type() (in module tart.utils.model_utils) GraphDataset (class in tart.representation.dataset)

H

half() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

has_download (tart.representation.dataset.GraphDataset property) has_process (tart.representation.dataset.GraphDataset property)

I

index_select() (tart.representation.dataset.GraphDataset method) indices() (tart.representation.dataset.GraphDataset method) init_logger() (in module tart.utils.train_utils)

init_user_configs() (in module tart.representation.config) ipu() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

J

jittable() (tart.representation.models.WeightedGINConv method)

L

len() (tart.representation.dataset.GraphDataset method) load_search_space() (in module tart.inference.predict)

load_state_dict() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

M

make_tunable() (in module tart.representation.config) make_validation_set() (in module tart.utils.train_utils) message() (tart.representation.models.WeightedGINConv method) message_and_aggregate() (tart.representation.models.WeightedGINConv method) module tart tart.inference tart.inference.embed tart.inference.predict tart.representation tart.representation.config tart.representation.dataset tart.representation.encoders tart.representation.models tart.representation.test tart.representation.train tart.utils tart.utils.config_utils tart.utils.graph_utils tart.utils.infer_utils tart.utils.model_utils tart.utils.tart_utils tart.utils.train_utils

modules() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) my_collate() (in module tart.representation.dataset)

N

named_buffers() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) named_children() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) named_modules() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) named_parameters() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

num_classes (tart.representation.dataset.GraphDataset property) num_edge_attributes (tart.representation.dataset.GraphDataset property) num_edge_features (tart.representation.dataset.GraphDataset property) num_edge_labels (tart.representation.dataset.GraphDataset property) num_features (tart.representation.dataset.GraphDataset property) num_node_attributes (tart.representation.dataset.GraphDataset property) num_node_features (tart.representation.dataset.GraphDataset property) num_node_labels (tart.representation.dataset.GraphDataset property)

P

parameters() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) pre_transform() (tart.representation.dataset.GraphDataset method) precision() (in module tart.representation.test) predict() (tart.representation.models.SubgraphEmbedder method) predict_neighs_batched() (in module tart.inference.predict)

predictv2() (tart.representation.models.SubgraphEmbedder method) print_header() (in module tart.utils.tart_utils) print_summary() (tart.representation.dataset.GraphDataset method) process() (tart.representation.dataset.GraphDataset method) processed_dir (tart.representation.dataset.GraphDataset property) processed_file_names (tart.representation.dataset.GraphDataset property) processed_paths (tart.representation.dataset.GraphDataset property) propagate() (tart.representation.models.WeightedGINConv method)

R

random_samples_generator() (tart.representation.dataset.GraphDataset method) raw_dir (tart.representation.dataset.GraphDataset property) raw_file_names (tart.representation.dataset.GraphDataset property) raw_paths (tart.representation.dataset.GraphDataset property) read_embedding() (in module tart.inference.predict) read_graph_from_json() (in module tart.utils.graph_utils) recall() (in module tart.representation.test) register_aggregate_forward_hook() (tart.representation.models.WeightedGINConv method) register_aggregate_forward_pre_hook() (tart.representation.models.WeightedGINConv method) register_backward_hook() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) register_buffer() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) register_edge_update_forward_hook() (tart.representation.models.WeightedGINConv method) register_edge_update_forward_pre_hook() (tart.representation.models.WeightedGINConv method) register_forward_hook() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) register_forward_pre_hook() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

register_full_backward_hook() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) register_load_state_dict_post_hook() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) register_message_and_aggregate_forward_hook() (tart.representation.models.WeightedGINConv method) register_message_and_aggregate_forward_pre_hook() (tart.representation.models.WeightedGINConv method) register_message_forward_hook() (tart.representation.models.WeightedGINConv method) register_message_forward_pre_hook() (tart.representation.models.WeightedGINConv method) register_module() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) register_parameter() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) register_propagate_forward_hook() (tart.representation.models.WeightedGINConv method) register_propagate_forward_pre_hook() (tart.representation.models.WeightedGINConv method) requires_grad_() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) reset_parameters() (tart.representation.models.WeightedGINConv method)

S

sample_neigh() (tart.representation.dataset.GraphDataset method) search_space_sample() (in module tart.inference.predict) set_extra_state() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) set_parser() (in module tart.utils.train_utils) share_memory() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

shuffle() (tart.representation.dataset.GraphDataset method) start_workers() (in module tart.utils.train_utils) start_workers_embed() (in module tart.inference.embed) start_workers_process() (in module tart.inference.embed) state_dict() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) SubgraphEmbedder (class in tart.representation.models) summarize_tart_run() (in module tart.utils.tart_utils)

T

tart module tart.inference module tart.inference.embed module tart.inference.predict module tart.representation module tart.representation.config module tart.representation.dataset module tart.representation.encoders module tart.representation.models module tart.representation.test module tart.representation.train module tart.utils module tart.utils.config_utils module tart.utils.graph_utils module

tart.utils.infer_utils module tart.utils.model_utils module tart.utils.tart_utils module tart.utils.train_utils module tart_embed() (in module tart.inference.embed) tart_predict() (in module tart.inference.predict) tart_test() (in module tart.representation.test) tart_train() (in module tart.representation.train) test() (in module tart.representation.test) to() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) to_datapipe() (tart.representation.dataset.GraphDataset method) to_empty() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) train() (in module tart.representation.train) (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method) train_loop() (in module tart.representation.train) type() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

U

update() (tart.representation.models.WeightedGINConv method)

V

validate_feat_encoder() (in module tart.utils.config_utils)

validation() (in module tart.representation.test)

W

WeightedGINConv (class in tart.representation.models)

X

xpu() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)

Z

zero_grad() (tart.representation.models.BasicGNN method) (tart.representation.models.SubgraphEmbedder method) (tart.representation.models.WeightedGINConv method)