MCPcopy Create free account

hub / github.com/HKUDS/AI-Researcher / functions

Functions951 in github.com/HKUDS/AI-Researcher

Method__init__
Initialize the graph learning model. Args: input_dim (int): Input feature dimension hidden_dim (int): Hidden
examples/gnn_nodeformer/project/model/graph_model.py:37
Method__init__
(self, optimizer, warmup_epochs, total_epochs, max_lr, min_lr)
examples/gnn_nodeformer/project/experiments/run_enhanced_experiments.py:32
Method__init__
(self, config)
examples/fsq/project/run_adaptive_quantization.py:16
Method__init__
(self, base_levels=5, level_multiplier=2, num_hierarchies=3, embedding_dim=32)
examples/fsq/project/run_hierarchical_quantization.py:18
Method__init__
(self, beta=0.25, entropy_weight=0.1)
examples/fsq/project/run_regularized_training.py:18
Method__init__
Decoder network for VAE. Args: out_channels (int): Number of output channels latent_dim (int): Dimension of l
examples/fsq/project/model/vae.py:46
Method__init__
VAE with Finite Scalar Quantization. Args: in_channels (int): Number of input channels latent_dim (int): Dime
examples/fsq/project/model/vae.py:84
Method__init__
Initialize Finite Scalar Quantization module. Args: num_levels (int): Number of quantization levels embedding_dim (in
examples/fsq/project/model/quantization.py:7
Method__init__
Initialize trainer. Args: model: Model to train train_loader: Training data loader test_loader: T
examples/fsq/project/training/trainer.py:6
Method__init__
CIFAR-10 dataset with preprocessing. Args: root_dir (str): Directory containing dataset train (bool): If True
examples/fsq/project/data_processing/dataset.py:8
Method__init__
Initialize Inception Score calculator. Args: device: Device to run calculations on
examples/fsq/project/testing/evaluator.py:9
Method__init__
Initialize evaluator. Args: model: Model to evaluate test_loader: Test data loader device: Device
examples/fsq/project/testing/evaluator.py:83
Method__init__
(self, input_dim=3072, hidden_dim=512)
examples/con_flowmatching/project/model/network.py:6
Method__init__
(self, input_dim=3072, hidden_dims=None, activation='relu')
examples/con_flowmatching/project/model/velocity_network.py:6
Method__init__
(self, hidden_dims=[128, 256, 256, 128], activation='relu')
examples/con_flowmatching/project/model/resnet_velocity.py:24
Method__init__
(self, velocity_net, sigma_min=0.1, n_steps=100)
examples/con_flowmatching/project/model/resnet_velocity.py:82
Method__init__
(self, model, train_loader, test_loader, device, lr=2e-4, weight_decay=1e-4, ema_decay=0.999
examples/con_flowmatching/project/training/trainer.py:13
Method__init__
(self, dataloader)
examples/con_flowmatching/project/data_processing/cifar10.py:46
Method__init__
(self, device)
examples/con_flowmatching/project/testing/evaluator.py:9
Method__init__
(self, device, splits=10)
examples/con_flowmatching/project/testing/scores.py:8
Method__init__
(self, n_neighbors=5, eps=1e-8)
examples/con_flowmatching/project/testing/scores.py:53
Method__init__
(self, model, log_dir)
examples/rotation_vq/project/run_visualization_experiments.py:13
Method__init__
(self, k_dim=1024, z_dim=256, beta=0.25)
examples/rotation_vq/project/model/vqvae.py:25
Method__init__
(self, model, train_loader, test_loader, device, lr=2e-4)
examples/rotation_vq/project/training/trainer.py:9
Method__init__
(self, data_dir, train=True, transform=None)
examples/rotation_vq/project/data_processing/cifar10.py:13
Method__init__
Initialize VQ-VAE model evaluator Args: model (nn.Module): Trained VQ-VAE model test_loader (DataLoa
examples/rotation_vq/project/testing/evaluator.py:6
Method__init__
(self, hidden_dim)
examples/gnn_difformer/project/model/diffusion.py:7
Method__init__
(self, hidden_dim, tau=0.1)
examples/gnn_difformer/project/model/diffusion.py:54
Method__init__
(self, input_dim, hidden_dim, num_classes, num_layers=2, tau=0.1, lambda_reg=1.0)
examples/gnn_difformer/project/model/diffusion.py:80
Method__init__
(self, hidden_dim, num_heads=4, dropout=0.1)
examples/gnn_difformer/project/model/improved_diffusion.py:8
Method__init__
(self, hidden_dim, tau=0.1, dropout=0.1)
examples/gnn_difformer/project/model/improved_diffusion.py:79
Method__init__
(self, input_dim, hidden_dim, num_classes, num_layers=2, tau=0.1, lambda_reg=1.0, dropout=0.
examples/gnn_difformer/project/model/improved_diffusion.py:106
Method__init__
(self, user_num, item_num, embed_dim, n_layers=2)
examples/hgcl/project/model/heterogeneous_gnn.py:7
Method__init__
(self, input_dim, hidden_dim, output_dim, rank=10)
examples/hgcl/project/model/meta_network.py:6
Method__init__
(self, user_num, item_num, embed_dim, n_layers=2, temp=0.1, lambda_coef=0.5)
examples/hgcl/project/model/contrastive_model.py:8
Method__init__
(self, model, dataset, learning_rate=0.001, device='cuda')
examples/hgcl/project/training/trainer.py:7
Method__init__
(self, train_mat, batch_size=2048, num_negatives=1)
examples/hgcl/project/data_processing/dataset.py:58
Method__init__
(self, cache_path: str, log_path: Union[str, None, MetaChainLogger] = None, model: str = "gpt-4o-2024-08-06",
research_agent/run_infer_idea.py:101
Method__init__
(self, cache_path: str, log_path: Union[str, None, MetaChainLogger] = None, model: str = "gpt-4o-2024-08-06",
research_agent/run_infer_plan.py:102
Method__init__
(self, message)
research_agent/inno/fn_call_converter.py:31
Method__init__
log_path: path of log file, None
research_agent/inno/core.py:97
Method__init__
(self, log_path: str)
research_agent/inno/logger.py:10
Method__init__
(self, users: List[str])
research_agent/inno/util.py:402
Method__init__
Initialize the GitHub client Args: token: GitHub Personal Access Token, if None, try to get from environment var
research_agent/inno/tools/github_client.py:9
Method__init__
Initialize the GitHub searcher Args: token: GitHub Personal Access Token, optional
research_agent/inno/tools/github_client.py:71
Method__init__
(self, model: str)
research_agent/inno/memory/code_memory.py:58
Method__init__
Memory: memory and external knowledge management. Args: project_path: the path to store the data. embedding_m
research_agent/inno/memory/rag_memory.py:15
Method__init__
(self, model: str)
research_agent/inno/memory/rag_memory.py:175
Method__init__
( self, project_path: str, db_name: str = '.tool_table', platform: str = 'Open
research_agent/inno/memory/tool_memory.py:12
Method__init__
(self, project_path: str, db_name: str = '.code_tree', platform: str = 'OpenAI', api_key: str = None, embeddin
research_agent/inno/memory/codetree_memory.py:12
Method__init__
( self, project_path: str, db_name: str = '.paper_table', platform: str = 'Ope
research_agent/inno/memory/paper_memory.py:11
Method__init__
(self, language: str, node_types: list[str], path_to_object_file: str)
research_agent/inno/memory/code_tree/code_parser.py:22
Method__init__
(self, message='Failed to initialize browser environment')
research_agent/inno/environment/browser_env.py:43
Method__init__
(self, config: Union[DockerConfig, Dict])
research_agent/inno/environment/docker_env.py:28
Method__init__
(self, **options: Any)
research_agent/inno/environment/mdconvert.py:68
Method__init__
(self, title: Union[str, None] = None, text_content: str = "")
research_agent/inno/environment/mdconvert.py:136
Method__init__
Perform a Bing web search, and return the results formatted in Markdown. Args: bing_api_key: key for the Bing search API
research_agent/inno/environment/markdown_browser/markdown_search.py:31
Method__init__
Instantiate a new RequestsMarkdownBrowser. Arguments: start_page: The page on which the browser starts (default: "about:
research_agent/inno/environment/markdown_browser/requests_markdown_browser.py:76
Method__init__
(self, **options: Any)
research_agent/inno/environment/markdown_browser/mdconvert.py:58
Method__init__
(self, title: Union[str, None] = None, text_content: str = "")
research_agent/inno/environment/markdown_browser/mdconvert.py:126
Method__init__
(self)
research_agent/inno/environment/markdown_browser/abstract_markdown_browser.py:20
Method__init__
(self, agent: Agent, client: MetaChain, cache_path: str)
research_agent/inno/workflow/flowcache.py:11
Method__init__
(self, tool: Callable[[Any], Union[str, Dict]], cache_path: str)
research_agent/inno/workflow/flowcache.py:72
Method__init__
(self, cache_path: str, log_path: Union[str, None, MetaChainLogger] = None, model: str = "gpt-4o-2024-08-06")
research_agent/inno/workflow/flowcache.py:109
Method__init__
(self)
research_agent/inno/workflow/flowgraph.py:8
Method__init__
(self, research_field: str, structure_iterations: int = 3)
paper_agent/methodology_composing_using_template.py:14
Method__init__
(self, research_field: str, structure_iterations: int = 3)
paper_agent/introduction_composing.py:12
Method__init__
(self, research_field: str, section_name: str, structure_iterations: int = 3, gpt_model='gpt-4o-mini-2024-07-1
paper_agent/section_composer.py:27
Method__init__
(self, research_field: str, structure_iterations: int = 2)
paper_agent/conclusion_composing.py:12
Method__init__
(self, research_field: str, structure_iterations: int = 3, gpt_model='gpt-4o-mini-2024-07-18')
paper_agent/experiments_composing.py:12
Method__init__
(self, research_field: str, structure_iterations: int = 2)
paper_agent/abstract_composing.py:13
Method__init__
(self, research_field: str, structure_iterations: int = 3)
paper_agent/related_work_composing_using_template.py:12
Method__iter__
(self)
examples/con_flowmatching/project/data_processing/cifar10.py:50
Method__len__
(self)
examples/dccf/project/data_processing/dataset.py:22
Method__len__
(self)
examples/fsq/project/data_processing/dataset.py:34
Method__len__
(self)
examples/con_flowmatching/project/data_processing/cifar10.py:54
Method__len__
(self)
examples/rotation_vq/project/data_processing/cifar10.py:32
Method__new__
(cls)
research_agent/inno/registry.py:35
Function_click_id
Clicks the mouse on the target with the given element bid. Examples: _click_id('12') _click_id('12', button='left')
research_agent/inno/environment/browser_env.py:170
Function_get_page_markdown
Get the markdown content of the current page Examples: _get_page_markdown()
research_agent/inno/environment/mdconvert.py:4
Method_get_single_content
获取单个文件的内容
research_agent/inno/tools/github_client.py:214
Method_init_weights
(self, module)
examples/con_flowmatching/project/model/velocity_network.py:40
Method_local_to_docker
Convert a local path to a docker path local_path: the local path to convert, like `{local_workplace}/downloads/xxx`
research_agent/inno/environment/browser_env.py:427
Method_processFacts
(elm: List[Dict[str, Any]])
research_agent/inno/environment/markdown_browser/markdown_search.py:85
Function_visit_page
Visit a page, including downloading files based on the url Examples: _visit_page('https://archive.org/download/higpt_stage2/instruct
research_agent/inno/environment/browser_env.py:58
Methodaddress
Return the address of the current page.
research_agent/inno/environment/markdown_browser/requests_markdown_browser.py:134
Methodaddress
(self)
research_agent/inno/environment/markdown_browser/abstract_markdown_browser.py:25
Functionagent
Run an agent with a given model, agent function, query, and context variables. Args: model (str): The name of the model. agen
research_agent/inno/cli.py:15
Methodagents
(self)
research_agent/inno/registry.py:95
Methodagents_info
(self)
research_agent/inno/registry.py:103
Functionask_text
Display a question in a panel and prompt the user for an answer. :param question: the question to display. :param title: the title of the
research_agent/inno/util.py:58
Functionasync_sleep_if_should_continue
(timeout: float)
research_agent/inno/environment/shutdown_listener.py:59
Methodbrowser_process
(self)
research_agent/inno/environment/browser_env.py:409
Functioncase_not_resolved
Use this function when the case is not resolved when all agents have tried their best. [IMPORTANT] Please do not use this function unless all
research_agent/inno/tools/inner.py:16
Functioncase_not_resolved
The failure reason is the reason why you cannot find a solution to the task. You can use this function only after you have tried multiple times an
research_agent/inno/agents/inno_agent/ml_agent.py:18
Functioncase_resolved
The task response is the result of the task. Use this function only after you have successfully completed the task. Args: task_response
research_agent/inno/agents/inno_agent/ml_agent.py:9
Functioncase_resolved
Use this function to given the analysis report of exsiting experiments and the further plan to the `Machine Learning Agent` to do more experiment
research_agent/inno/agents/inno_agent/exp_analyser.py:20
Functioncase_resolved
After you have taken enough notes for the innovation, you should use this function to merge the notes for the further innovation.
research_agent/inno/agents/inno_agent/survey_agent.py:144
Functioncase_resolved
The function to output the determined reference codebases. Use this function only after you have carefully reviewed the existing resources and un
research_agent/inno/agents/inno_agent/prepare_agent.py:10
Functioncase_resolved
Use this function when you have finished the task and want to give a suggestion about the implementation. You can only use this function after yo
research_agent/inno/agents/inno_agent/judge_agent.py:25
← previousnext →601–700 of 951, ranked by callers