API Documentation

APIs

• tart.inference
  • tart.inference.embed module
    • embed_main()
    • generate_embeddings()
    • generate_neighborhoods()
    • get_neighborhoods()
    • start_workers_embed()
    • start_workers_process()
    • tart_embed()
  • tart.inference.predict module
    • load_search_space()
    • predict_neighs_batched()
    • read_embedding()
    • search_space_sample()
    • tart_predict()
  • Module contents
• tart.representation
  • tart.representation.config module
    • build_feature_configs()
    • build_model_configs()
    • build_optimizer_configs()
    • init_user_configs()
    • make_tunable()
  • tart.representation.dataset module
    • Corpus
      • Corpus.gen_data_loader()
    • GraphDataset
      • GraphDataset.download()
      • GraphDataset.get()
      • GraphDataset.get_summary()
      • GraphDataset.has_download
      • GraphDataset.has_process
      • GraphDataset.index_select()
      • GraphDataset.indices()
      • GraphDataset.len()
      • GraphDataset.num_classes
      • GraphDataset.num_edge_attributes
      • GraphDataset.num_edge_features
      • GraphDataset.num_edge_labels
      • GraphDataset.num_features
      • GraphDataset.num_node_attributes
      • GraphDataset.num_node_features
      • GraphDataset.num_node_labels
      • GraphDataset.pre_transform()
      • GraphDataset.print_summary()
      • GraphDataset.process()
      • GraphDataset.processed_dir
      • GraphDataset.processed_file_names
      • GraphDataset.processed_paths
      • GraphDataset.random_samples_generator()
      • GraphDataset.raw_dir
      • GraphDataset.raw_file_names
      • GraphDataset.raw_paths
      • GraphDataset.sample_neigh()
      • GraphDataset.shuffle()
      • GraphDataset.to_datapipe()
    • my_collate()
  • tart.representation.encoders module
    • codebert_bpe_encoder()
    • codebert_encoder()
    • get_feature_encoder()
  • tart.representation.models module
    • BasicGNN
      • BasicGNN.T_destination
      • BasicGNN.add_module()
      • BasicGNN.apply()
      • BasicGNN.bfloat16()
      • BasicGNN.buffers()
      • BasicGNN.children()
      • BasicGNN.cpu()
      • BasicGNN.cuda()
      • BasicGNN.double()
      • BasicGNN.dump_patches
      • BasicGNN.eval()
      • BasicGNN.extra_repr()
      • BasicGNN.float()
      • BasicGNN.forward()
      • BasicGNN.get_agg_layer()
      • BasicGNN.get_buffer()
      • BasicGNN.get_extra_state()
      • BasicGNN.get_parameter()
      • BasicGNN.get_submodule()
      • BasicGNN.half()
      • BasicGNN.ipu()
      • BasicGNN.load_state_dict()
      • BasicGNN.loss()
      • BasicGNN.modules()
      • BasicGNN.named_buffers()
      • BasicGNN.named_children()
      • BasicGNN.named_modules()
      • BasicGNN.named_parameters()
      • BasicGNN.parameters()
      • BasicGNN.register_backward_hook()
      • BasicGNN.register_buffer()
      • BasicGNN.register_forward_hook()
      • BasicGNN.register_forward_pre_hook()
      • BasicGNN.register_full_backward_hook()
      • BasicGNN.register_load_state_dict_post_hook()
      • BasicGNN.register_module()
      • BasicGNN.register_parameter()
      • BasicGNN.requires_grad_()
      • BasicGNN.set_extra_state()
      • BasicGNN.share_memory()
      • BasicGNN.state_dict()
      • BasicGNN.to()
      • BasicGNN.to_empty()
      • BasicGNN.train()
      • BasicGNN.training
      • BasicGNN.type()
      • BasicGNN.xpu()
      • BasicGNN.zero_grad()
    • Preprocess
      • Preprocess.T_destination
      • Preprocess.add_module()
      • Preprocess.apply()
      • Preprocess.bfloat16()
      • Preprocess.buffers()
      • Preprocess.children()
      • Preprocess.cpu()
      • Preprocess.cuda()
      • Preprocess.dim_out
      • Preprocess.double()
      • Preprocess.dump_patches
      • Preprocess.eval()
      • Preprocess.extra_repr()
      • Preprocess.float()
      • Preprocess.forward()
      • Preprocess.get_buffer()
      • Preprocess.get_extra_state()
      • Preprocess.get_parameter()
      • Preprocess.get_submodule()
      • Preprocess.half()
      • Preprocess.ipu()
      • Preprocess.load_state_dict()
      • Preprocess.modules()
      • Preprocess.named_buffers()
      • Preprocess.named_children()
      • Preprocess.named_modules()
      • Preprocess.named_parameters()
      • Preprocess.parameters()
      • Preprocess.register_backward_hook()
      • Preprocess.register_buffer()
      • Preprocess.register_forward_hook()
      • Preprocess.register_forward_pre_hook()
      • Preprocess.register_full_backward_hook()
      • Preprocess.register_load_state_dict_post_hook()
      • Preprocess.register_module()
      • Preprocess.register_parameter()
      • Preprocess.requires_grad_()
      • Preprocess.set_extra_state()
      • Preprocess.share_memory()
      • Preprocess.state_dict()
      • Preprocess.to()
      • Preprocess.to_empty()
      • Preprocess.train()
      • Preprocess.training
      • Preprocess.type()
      • Preprocess.xpu()
      • Preprocess.zero_grad()
    • SubgraphEmbedder
      • SubgraphEmbedder.T_destination
      • SubgraphEmbedder.add_module()
      • SubgraphEmbedder.apply()
      • SubgraphEmbedder.bfloat16()
      • SubgraphEmbedder.buffers()
      • SubgraphEmbedder.children()
      • SubgraphEmbedder.cpu()
      • SubgraphEmbedder.criterion()
      • SubgraphEmbedder.cuda()
      • SubgraphEmbedder.double()
      • SubgraphEmbedder.dump_patches
      • SubgraphEmbedder.eval()
      • SubgraphEmbedder.extra_repr()
      • SubgraphEmbedder.float()
      • SubgraphEmbedder.forward()
      • SubgraphEmbedder.get_buffer()
      • SubgraphEmbedder.get_extra_state()
      • SubgraphEmbedder.get_parameter()
      • SubgraphEmbedder.get_submodule()
      • SubgraphEmbedder.half()
      • SubgraphEmbedder.ipu()
      • SubgraphEmbedder.load_state_dict()
      • SubgraphEmbedder.modules()
      • SubgraphEmbedder.named_buffers()
      • SubgraphEmbedder.named_children()
      • SubgraphEmbedder.named_modules()
      • SubgraphEmbedder.named_parameters()
      • SubgraphEmbedder.parameters()
      • SubgraphEmbedder.predict()
      • SubgraphEmbedder.predictv2()
      • SubgraphEmbedder.register_backward_hook()
      • SubgraphEmbedder.register_buffer()
      • SubgraphEmbedder.register_forward_hook()
      • SubgraphEmbedder.register_forward_pre_hook()
      • SubgraphEmbedder.register_full_backward_hook()
      • SubgraphEmbedder.register_load_state_dict_post_hook()
      • SubgraphEmbedder.register_module()
      • SubgraphEmbedder.register_parameter()
      • SubgraphEmbedder.requires_grad_()
      • SubgraphEmbedder.set_extra_state()
      • SubgraphEmbedder.share_memory()
      • SubgraphEmbedder.state_dict()
      • SubgraphEmbedder.to()
      • SubgraphEmbedder.to_empty()
      • SubgraphEmbedder.train()
      • SubgraphEmbedder.training
      • SubgraphEmbedder.type()
      • SubgraphEmbedder.xpu()
      • SubgraphEmbedder.zero_grad()
    • WeightedGINConv
      • WeightedGINConv.T_destination
      • WeightedGINConv.add_module()
      • WeightedGINConv.aggregate()
      • WeightedGINConv.apply()
      • WeightedGINConv.bfloat16()
      • WeightedGINConv.buffers()
      • WeightedGINConv.children()
      • WeightedGINConv.cpu()
      • WeightedGINConv.cuda()
      • WeightedGINConv.double()
      • WeightedGINConv.dump_patches
      • WeightedGINConv.edge_update()
      • WeightedGINConv.edge_updater()
      • WeightedGINConv.eval()
      • WeightedGINConv.explain
      • WeightedGINConv.explain_message()
      • WeightedGINConv.extra_repr()
      • WeightedGINConv.float()
      • WeightedGINConv.forward()
      • WeightedGINConv.get_buffer()
      • WeightedGINConv.get_extra_state()
      • WeightedGINConv.get_parameter()
      • WeightedGINConv.get_submodule()
      • WeightedGINConv.half()
      • WeightedGINConv.ipu()
      • WeightedGINConv.jittable()
      • WeightedGINConv.load_state_dict()
      • WeightedGINConv.message()
      • WeightedGINConv.message_and_aggregate()
      • WeightedGINConv.modules()
      • WeightedGINConv.named_buffers()
      • WeightedGINConv.named_children()
      • WeightedGINConv.named_modules()
      • WeightedGINConv.named_parameters()
      • WeightedGINConv.parameters()
      • WeightedGINConv.propagate()
      • WeightedGINConv.register_aggregate_forward_hook()
      • WeightedGINConv.register_aggregate_forward_pre_hook()
      • WeightedGINConv.register_backward_hook()
      • WeightedGINConv.register_buffer()
      • WeightedGINConv.register_edge_update_forward_hook()
      • WeightedGINConv.register_edge_update_forward_pre_hook()
      • WeightedGINConv.register_forward_hook()
      • WeightedGINConv.register_forward_pre_hook()
      • WeightedGINConv.register_full_backward_hook()
      • WeightedGINConv.register_load_state_dict_post_hook()
      • WeightedGINConv.register_message_and_aggregate_forward_hook()
      • WeightedGINConv.register_message_and_aggregate_forward_pre_hook()
      • WeightedGINConv.register_message_forward_hook()
      • WeightedGINConv.register_message_forward_pre_hook()
      • WeightedGINConv.register_module()
      • WeightedGINConv.register_parameter()
      • WeightedGINConv.register_propagate_forward_hook()
      • WeightedGINConv.register_propagate_forward_pre_hook()
      • WeightedGINConv.requires_grad_()
      • WeightedGINConv.reset_parameters()
      • WeightedGINConv.set_extra_state()
      • WeightedGINConv.share_memory()
      • WeightedGINConv.special_args
      • WeightedGINConv.state_dict()
      • WeightedGINConv.to()
      • WeightedGINConv.to_empty()
      • WeightedGINConv.train()
      • WeightedGINConv.training
      • WeightedGINConv.type()
      • WeightedGINConv.update()
      • WeightedGINConv.xpu()
      • WeightedGINConv.zero_grad()
  • tart.representation.test module
    • precision()
    • recall()
    • tart_test()
    • test()
    • validation()
  • tart.representation.train module
    • tart_train()
    • train()
    • train_loop()
  • Module contents
• tart.utils
  • tart.utils.config_utils module
    • validate_feat_encoder()
  • tart.utils.graph_utils module
    • featurize_graph()
    • read_graph_from_json()
  • tart.utils.infer_utils module
  • tart.utils.model_utils module
    • build_model()
    • build_optimizer()
    • get_device()
    • get_torch_tensor_type()
  • tart.utils.tart_utils module
    • print_header()
    • summarize_tart_run()
  • tart.utils.train_utils module
    • init_logger()
    • make_validation_set()
    • set_parser()
    • start_workers()
  • Module contents

Module contents